change output to text

app.py

@@ -37,8 +37,9 @@ if __name__ == '__main__':
 
     def predict(image):
         img = PILImage.create(image)
-        pred,pred_idx,probs = model.predict(img)
-        return {'Cat':probs[1].item(), 'Not Cat':probs[0].item()}
+        _,_,probs = model.predict(img)
+        return {'Not Cat':float("{:.2f}".format(probs[0].item())), 'Cat':float("{:.2f}".format(probs[1].item()))}
 
-    demo = gr.Interface(fn=model.predict, inputs="image", outputs=['label'],examples=['examples/cat-1.jpg','examples/dog-1.jpg'],allow_flagging='never')
-    demo.launch()
+    # outputs label is not working, need to investigate further
+    demo = gr.Interface(fn=model.predict, inputs=gr.Image(), outputs='text',examples=['examples/cat-1.jpg','examples/dog-1.jpg'],allow_flagging='never')
+    demo.launch(show_error=True)