Shriharsh commited on
Commit
8305287
โ€ข
1 Parent(s): 6256012

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -50,8 +50,9 @@ def predict(img) -> Tuple[Dict, float]:
50
 
51
  # Create title, description and article
52
  title = "FoodVision BIG ๐Ÿ”๐Ÿ‘๐Ÿ’ช"
53
- description = "An [EfficientNetB2 feature extractor](https://pytorch.org/vision/stable/models/generated/torchvision.models.efficientnet_b2.html#torchvision.models.efficientnet_b2) computer vision model to classify images [101 classes of food from the Food101 dataset](https://github.com/mrdbourke/pytorch-deep-learning/blob/main/extras/food101_class_names.txt)."
54
- article = "Created at [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/#11-turning-our-foodvision-big-model-into-a-deployable-app)."
 
55
 
56
  # Create example list
57
  example_list = [["examples/" + example] for example in os.listdir("examples")]
 
50
 
51
  # Create title, description and article
52
  title = "FoodVision BIG ๐Ÿ”๐Ÿ‘๐Ÿ’ช"
53
+ description = "An [EfficientNetB2 feature extractor]
54
+ #(https://pytorch.org/vision/stable/models/generated/torchvision.models.efficientnet_b2.html#torchvision.models.efficientnet_b2) computer vision model to classify images [101 classes of food from the Food101 dataset](https://github.com/mrdbourke/pytorch-deep-learning/blob/main/extras/food101_class_names.txt)."
55
+ #article = "Created at [09. PyTorch Model Deployment](https://www.learnpytorch.io/09_pytorch_model_deployment/#11-turning-our-foodvision-big-model-into-a-deployable-app)."
56
 
57
  # Create example list
58
  example_list = [["examples/" + example] for example in os.listdir("examples")]