Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -11,6 +11,8 @@ labels = learn.dls.vocab
|
|
| 11 |
def classify_trash(img):
|
| 12 |
pred_class, pred_idx, probs = learn.predict(img)
|
| 13 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
|
|
|
|
|
|
| 14 |
|
| 15 |
# Gradio Interface
|
| 16 |
interface = gr.Interface(
|
|
@@ -18,7 +20,8 @@ interface = gr.Interface(
|
|
| 18 |
inputs=gr.Image(type="pil"),
|
| 19 |
outputs=gr.Label(num_top_classes=5),
|
| 20 |
title="Trash Classifier",
|
| 21 |
-
description="Upload a trash image to classify it into one of 5 categories."
|
|
|
|
| 22 |
)
|
| 23 |
|
| 24 |
# Launch app
|
|
|
|
| 11 |
def classify_trash(img):
|
| 12 |
pred_class, pred_idx, probs = learn.predict(img)
|
| 13 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
| 14 |
+
|
| 15 |
+
examples = ["glass.png","plastic.jpg"]
|
| 16 |
|
| 17 |
# Gradio Interface
|
| 18 |
interface = gr.Interface(
|
|
|
|
| 20 |
inputs=gr.Image(type="pil"),
|
| 21 |
outputs=gr.Label(num_top_classes=5),
|
| 22 |
title="Trash Classifier",
|
| 23 |
+
description="Upload a trash image to classify it into one of 5 categories.",
|
| 24 |
+
examples=examples
|
| 25 |
)
|
| 26 |
|
| 27 |
# Launch app
|