fix: label issue

app.py

@@ -38,8 +38,8 @@ if __name__ == '__main__':
     def predict(image):
         img = PILImage.create(image)
         _,_,probs = model.predict(img)
-        return {'Not Cat':float("{:.2f}".format(probs[0].item())), 'Cat':float("{:.2f}".format(probs[1].item()))}
+        return {'Not a Cat':float("{:.2f}".format(probs[0].item())), 'Cat':float("{:.2f}".format(probs[1].item()))}
 
     # 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 = gr.Interface(fn=predict, inputs=gr.Image(), outputs='label',examples=['examples/cat-1.jpg','examples/dog-1.jpg'],allow_flagging='never')
     demo.launch(show_error=True)