pritesh29 commited on
Commit
e1af374
Β·
1 Parent(s): 6865873

added food labels and training code

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -57,9 +57,9 @@ def predict(img) -> tuple[dict, float]:
57
  ### 4. Gradio app
58
 
59
  # Create title, description and article strings
60
- title = "FoodVision πŸ”πŸ‘"
61
- description = "An EfficientNetB2 feature extractor Computer Vision model to classify images of Food into 101 different classes."
62
- article = "Created by [pritesh tadvi](https://pritesh2000.github.io/)"
63
 
64
  # Create examples list from "examples/" directory
65
  example_list = [["examples/"+ example] for example in os.listdir("examples")]
 
57
  ### 4. Gradio app
58
 
59
  # Create title, description and article strings
60
+ title = "FoodVision πŸ˜‹"
61
+ description = "An EfficientNetB2 feature extractor Computer Vision model to classify images of Food into [101 different classes](https://huggingface.co/spaces/pritesh29/FoodVision/blob/main/class_names.txt)."
62
+ article = "Created by [pritesh tadvi](https://pritesh2000.github.io/)\nTraining code at [kaggle/foodvision](https://www.kaggle.com/code/priteshtadvi/foodvision-a-food-classification-model)"
63
 
64
  # Create examples list from "examples/" directory
65
  example_list = [["examples/"+ example] for example in os.listdir("examples")]