Update app.py

app.py

@@ -17,12 +17,23 @@ categories = ('Dog', 'Cat')
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
-    return dict(zip(categories, map(float,probs)))
+    if probs[0]>probs[1]:
+        pred_class = 'This is Dog'
+    else:
+        pred_class = 'This is Cat'
+    return pred_class, dict(zip(categories, map(float,probs)))
 
 # Cell
 image = gr.inputs.Image(shape=(192, 192))
-label = gr.outputs.Label()
-examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
 
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+set_label = gr.outputs.Textbox(label="Predicted Class")
+
+set_prob = gr.outputs.Label(num_top_classes=2, label="Predicted Probability Per Class")
+examples = ['test1.jpg', 'test2.jpg', 'test3.jpeg', 'test4.jpeg', 'test5.jpeg', 'test6.jpeg', 'test7.jpeg', 'test8.jpeg', 'test9.jpeg', 'test10.jpeg']
+
+intf = gr.Interface(fn=classify_image, 
+                    inputs=image, 
+                    outputs=[set_label, set_prob], 
+                    examples_per_page = 2, 
+                    examples=examples)
 intf.launch(inline=False)