Fix: Convert boolean prediction to 'Cat' or 'Dog'

app.py

@@ -23,11 +23,14 @@ def predict(image):
 
         # Run prediction
         pred, _, probs = model.predict(image)
+
+        # Convert boolean prediction to "Cat" or "Dog"
+        label = "Cat" if pred else "Dog"
         confidence = float(probs.max())  # Convert Tensor to float
         
-        print(f"✅ Prediction successful: {pred}, Confidence: {confidence:.2f}")
+        print(f"✅ Prediction successful: {label}, Confidence: {confidence:.2f}")
         
-        return f"Prediction: {pred} (Confidence: {confidence:.2f})"
+        return f"Prediction: {label} (Confidence: {confidence:.2f})"
     
     except Exception as e:
         print(f"❌ Error during prediction: {e}")