Update app.py

app.py

@@ -8,10 +8,10 @@ labels = neural_net.dls.vocab
 
 def predict(img):
     # img = PILImage.create(img)
-    # category, idx, probs = neural_net.predict(img)
-    # return dict(zip(labels, map(float, probs)))
+    category, idx, probs = neural_net.predict(img)
+    return dict(zip(labels, map(float, probs)))
     # return {'cat': 0.3, 'dog': 0.7}
-    return img.rotate(45)
+    # return img.rotate(45)
     
 title = 'Natural Landscape Classifier'
 
@@ -19,7 +19,7 @@ examples = ['farm.png']
 
 iface = gr.Interface(fn=predict, 
                      inputs=gr.Image(type='pil'), 
-                     outputs='image',
+                     outputs='label',
                      title=title,
                      examples=examples)
 iface.launch(share=True)