thuyentruong commited on
Commit
971d574
·
verified ·
1 Parent(s): ef1d090

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
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="This image classification model was fine-tuned from vit-base-patch16-224-in21k by Google.<br> Model id: thuyentruong/food_classification_model.<br> Double-click on the example photo below to see the classification in action.",
 
 
 
 
 
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()