arturomacias commited on
Commit
023e2e8
·
1 Parent(s): 8d1b6b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -54,8 +54,7 @@ def predict(img) -> Tuple[Dict, float]:
54
 
55
  example_list = [["examples/" + example] for example in os.listdir("examples")]
56
  title = "Computer Vision for Food Processing"
57
- description = "An [EfficientNetB2](https://pytorch.org/vision/main/models/generated/torchvision.models.efficientnet_b2.html?highlight=efficientnet_b2#torchvision.models.efficientnet_b2) feature extractor computer vision model to classify images of 101 foods (taken from PyTorch's [Food101](https://pytorch.org/vision/main/generated/torchvision.datasets.Food101.html) dataset).
58
- View the foods that this model can classify [here](https://github.com/alpapado/food-101/blob/master/data/meta/classes.txt)."
59
 
60
  demo = gr.Interface(fn=predict,
61
  inputs=gr.Image(type="pil"),
 
54
 
55
  example_list = [["examples/" + example] for example in os.listdir("examples")]
56
  title = "Computer Vision for Food Processing"
57
+ description = "An [EfficientNetB2](https://pytorch.org/vision/main/models/generated/torchvision.models.efficientnet_b2.html?highlight=efficientnet_b2#torchvision.models.efficientnet_b2) feature extractor computer vision model to classify images of 101 foods (taken from PyTorch's [Food101](https://pytorch.org/vision/main/generated/torchvision.datasets.Food101.html) dataset). View the foods that this model can classify [here](https://github.com/alpapado/food-101/blob/master/data/meta/classes.txt)."
 
58
 
59
  demo = gr.Interface(fn=predict,
60
  inputs=gr.Image(type="pil"),