| import gradio as gr | |
| from transformers import pipeline | |
| pipe = pipeline(task="image-classification", | |
| model="thuyentruong/food_classification_model") | |
| gr.Interface.from_pipeline(pipe, | |
| title="Food Image Classification", | |
| description=""" | |
| Upload a food photo or double-click on the example photos below to see the classification in action. | |
| <br> Powered by a fine tuned Vision Transformer. | |
| <br> Base model: vit-base-patch16-224-in21k from Google. | |
| <br> Fine tune model: food_classification_model by Thuyen Truong. | |
| """, | |
| examples = ['beignets.jpeg','spagetti.jpeg'], | |
| ).launch() |