Update app.py
Browse files
app.py
CHANGED
@@ -5,6 +5,11 @@ pipe = pipeline(task="image-classification",
|
|
5 |
model="thuyentruong/food_classification_model")
|
6 |
gr.Interface.from_pipeline(pipe,
|
7 |
title="Food Image Classification",
|
8 |
-
description="
|
|
|
|
|
|
|
|
|
|
|
9 |
examples = ['beignets.jpeg','spagetti.jpeg'],
|
10 |
).launch()
|
|
|
5 |
model="thuyentruong/food_classification_model")
|
6 |
gr.Interface.from_pipeline(pipe,
|
7 |
title="Food Image Classification",
|
8 |
+
description="""
|
9 |
+
Upload a food photo or double-click on the example photos below to see the classification in action.
|
10 |
+
<br> Powered by a fine tuned Vision Transformer.
|
11 |
+
<br> Base model: vit-base-patch16-224-in21k from Google.
|
12 |
+
<br> Fine tune model: food_classification_model by Thuyen Truong.
|
13 |
+
""",
|
14 |
examples = ['beignets.jpeg','spagetti.jpeg'],
|
15 |
).launch()
|