Update app.py

app.py

@@ -31,6 +31,6 @@ def predict(img):
     otpt=F.softmax(learn.forward(inp_img.view(-1,64)))
     #pred,pred_idx,probs = learn.predict(img)
     
-    return {labels[i]: float(otpt[0].data[i]) for i in range(len(labels)),'image': inp_img}
+    return dict([labels[i]: float(otpt[0].data[i]) for i in range(len(labels))),inp_img
 
-gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)),   outputs=[gr.outputs.Label(num_top_classes=3), gr.outputs.Image()]).launch(share=True)
+gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)),   outputs=[gr.outputs.Label(num_top_classes=10), gr.outputs.Image()]).launch(share=True)