Fix: Convert image to Fastai PILImage before prediction

app.py

@@ -1,18 +1,27 @@
-from fastai.vision.all import load_learner
+from fastai.vision.all import load_learner, PILImage
 import gradio as gr
 from PIL import Image
 
-# Define the missing function
-def is_cat(x): 
-    return x[0].isupper()
-
 # Load the trained model
 model = load_learner('model.pkl')
 
 # Define a function to make predictions
 def predict(image):
-    pred, _, probs = model.predict(image)
-    return f"Prediction: {pred} (Confidence: {probs.max():.2f})"
+    try:
+        print("📸 Received image for prediction")
+
+        # Convert to Fastai's expected PILImage format
+        image = PILImage.create(image)
+
+        # Run prediction
+        pred, _, probs = model.predict(image)
+        print(f"✅ Prediction successful: {pred}, Confidence: {probs.max():.2f}")
+        
+        return f"Prediction: {pred} (Confidence: {probs.max():.2f})"
+    
+    except Exception as e:
+        print(f"❌ Error during prediction: {e}")
+        return f"Error: {e}"
 
 # Create the Gradio web interface
 interface = gr.Interface(