Update app.py

app.py

@@ -11,7 +11,12 @@ labels = ['Dog', 'Cat']
 def predict(img):
     img = PILImage.create(img)
     pred, pred_idx, probs = learn_inf.predict(img)
-    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+    
+    # Check if the prediction is "Cat" and return the appropriate string
+    if pred == 'Cat':
+        return "This is a cat"
+    else:
+        return "This is not a cat"
 
 title = "Is this image of a cat?"
 iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512,512)), outputs=gr.outputs.Label(num_top_classes=2), title=title)