Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -54,6 +54,8 @@ def predict(img) -> Tuple[Dict, float]:
|
|
| 54 |
return pred_labels_and_probs, pred_time
|
| 55 |
|
| 56 |
### 4. Gradio app ###
|
|
|
|
|
|
|
| 57 |
|
| 58 |
# Create Gradio interface
|
| 59 |
demo = gr.Interface(
|
|
@@ -63,6 +65,8 @@ demo = gr.Interface(
|
|
| 63 |
gr.Label(num_top_classes=5, label="Predictions"),
|
| 64 |
gr.Number(label="Prediction time (s)"),
|
| 65 |
],
|
|
|
|
|
|
|
| 66 |
)
|
| 67 |
|
| 68 |
# Launch the app!
|
|
|
|
| 54 |
return pred_labels_and_probs, pred_time
|
| 55 |
|
| 56 |
### 4. Gradio app ###
|
| 57 |
+
title = "Demo Of Group 29"
|
| 58 |
+
description = "A food classification model base on ViT"
|
| 59 |
|
| 60 |
# Create Gradio interface
|
| 61 |
demo = gr.Interface(
|
|
|
|
| 65 |
gr.Label(num_top_classes=5, label="Predictions"),
|
| 66 |
gr.Number(label="Prediction time (s)"),
|
| 67 |
],
|
| 68 |
+
title=title,
|
| 69 |
+
description=description,
|
| 70 |
)
|
| 71 |
|
| 72 |
# Launch the app!
|