Update app.py

app.py

@@ -1,11 +1,18 @@
 from fastai.vision.all import *
 import gradio as gr
 
-# Load the model and define categories
+# Define categories
 categories = ['dog', 'cat']
+
+# Define the function that was used during training
+def is_cat(x): 
+    return x[0].isupper()
+
+# Load the model
 learn = load_learner('model.pkl')
 
-def is_cat(img):
+# Gradio prediction function
+def predict_image(img):
     pred, idx, probs = learn.predict(img)
     return {categories[i]: float(probs[i]) for i in range(len(categories))}
 
@@ -16,7 +23,7 @@ examples = [['dog.jpg'], ['cat.jpg']]
 
 # Create and launch the interface
 interface = gr.Interface(
-    fn=is_cat,
+    fn=predict_image,
     inputs=image,
     outputs=label,
     examples=examples,