model, interface, example added

app.py

@@ -1,5 +1,4 @@
 from fastai.vision.all import *
-from fastai.vision.all import load_learner
 import gradio as gr
 
 # import pathlib
@@ -19,7 +18,7 @@ pollutant_labels = (
     "street litter"
 )
 
-model = load_learner('models/vispol-1-recognizer-v0.pkl')
+model = load_learner('vispol-1-recognizer-v0.pkl')
 
 def recognize_image(image):
   pred, idx, probs = model.predict(image)
@@ -27,13 +26,13 @@ def recognize_image(image):
 
 #!export
 image = gr.inputs.Image(shape=(256,256))
-label = gr.outputs.Label()
+label = gr.outputs.Label(num_top_classes=5)
 examples = [
-    'test_images/unknown_00.jpg',
-    'test_images/unknown_01.jpg',
-    'test_images/unknown_02.jpg',
-    'test_images/unknown_03.jpg',
-    'test_images/unknown_04.jpg'
+    'unknown_00.jpg',
+    'unknown_01.jpg',
+    'unknown_02.jpg',
+    'unknown_03.jpg',
+    'unknown_04.jpg'
     ]
 
 iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)