Update app.py

app.py

@@ -1,18 +1,19 @@
-import gradio as gr
-from transformers import pipeline
-
-MODEL_ID = "microsoft/resnet-50"
-clf = pipeline("image-classification", model=MODEL_ID)
-
-def predict(img):
-    out = clf(img)
-    # show top-3 with scores
-    out = sorted(out, key=lambda r: r["score"], reverse=True)[:3]
-    return {r["label"]: float(r["score"]) for r in out}
-
-gr.Interface(
-    fn=predict,
-    inputs=gr.Image(type="pil", label="Upload image"),
-    outputs=gr.Label(num_top_classes=3),
-    title="Image Classifier (pre-tuned)",
-).launch()
+import gradio as gr
+from transformers import pipeline
+
+MODEL_ID = "microsoft/resnet-50"
+clf = pipeline("image-classification", model=MODEL_ID)
+
+def predict(img):
+    out = clf(img)
+    # show top-3 with scores
+    out = sorted(out, key=lambda r: r["score"], reverse=True)[:3]
+    return {r["label"]: float(r["score"]) for r in out}
+
+gr.Interface(
+    fn=predict,
+    inputs=gr.Image(type="pil", label="Upload image"),
+    outputs=gr.Label(num_top_classes=3),
+    title="Image Classifier (pre-tuned)",
+    examples=[ ["banana-1.jpg"], ["cat1.jpg"], ["zebra.jpg"]  ]
+).launch()