Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 66 |
-
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()
|