Kshitij-369 commited on
Commit
b127f25
Β·
verified Β·
1 Parent(s): fec4a4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -49,10 +49,6 @@ def predict(img) -> Tuple[Dict, float]:
49
 
50
  ### 4. Gradio app ###
51
 
52
- # Create title, description and article
53
- title = "FoodVision Mini πŸ•πŸ₯©πŸ£"
54
- 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 as pizza, steak or sushi."
55
-
56
  # Create example list
57
  example_list = [["examples/" + example] for example in os.listdir("examples")]
58
 
@@ -62,8 +58,8 @@ demo = gr.Interface(fn=predict, # maps inputs to outputs
62
  outputs=[gr.Label(num_top_classes=3, label="Predictions"),
63
  gr.Number(label="Prediction time (s)")],
64
  examples=example_list,
65
- title=title,
66
- description=description)
67
 
68
  # Launch the demo!
69
  demo.launch()
 
49
 
50
  ### 4. Gradio app ###
51
 
 
 
 
 
52
  # Create example list
53
  example_list = [["examples/" + example] for example in os.listdir("examples")]
54
 
 
58
  outputs=[gr.Label(num_top_classes=3, label="Predictions"),
59
  gr.Number(label="Prediction time (s)")],
60
  examples=example_list,
61
+ title="FoodVision Mini πŸ•πŸ₯©πŸ£",
62
+ 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 as pizza, steak or sushi.")
63
 
64
  # Launch the demo!
65
  demo.launch()