Update app.py

app.py

@@ -10,17 +10,18 @@ inf = load_learner('export.pkl')
 
 def classify_img(img):
     pred, idx, probs = inf.predict(img)
-    return dict(zip(categories,map(float,probs)))
+    return pred
+    #return dict(zip(categories,map(float,probs)))
 
     #|export
 
 #This creates the gradio interface
 
 image = gr.Image(shape=(192,192))
-label = gr.outputs.Label()
+label = gr.outputs
 
 #examples = ['cheetah.jpg','leopard.jpg','img.jpg']
 
-intf = gr.Interface(fn = classify_img, inputs=image,outputs=label)
+intf = gr.Interface(fn = classify_img, inputs=image,outputs=pred)
 
 intf.launch()