feat: add examples

app.py

@@ -38,7 +38,7 @@ if __name__ == '__main__':
     def predict(image):
         img = PILImage.create(image)
         pred,pred_idx,probs = model.predict(img)
-        return f"Probability it's a cat: {probs[1].item():.6f}" 
+        return {'Cat':probs[1].item(), 'Not Cat':probs[0].item()}
 
-    demo = gr.Interface(fn=model.predict, inputs="image", outputs='text')
+    demo = gr.Interface(fn=model.predict, inputs="image", outputs=['label'],examples=['examples/cat-1.jpg','examples/dog-1.jpg'],allow_flagging='never')
     demo.launch()