Deploy Cat-Dog Recognition model

app.py

@@ -1,7 +1,23 @@
+from fastai.vision.all import load_learner
 import gradio as gr
+from PIL import Image
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load the trained model
+model = load_learner('model.pkl')
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+# Define a function to make predictions
+def predict(image):
+    pred, _, probs = model.predict(image)
+    return f"Prediction: {pred} (Confidence: {probs.max():.2f})"
+
+# Create the Gradio web interface
+interface = gr.Interface(
+    fn=predict,
+    inputs=gr.inputs.Image(type="pil"),
+    outputs="text",
+    title="Cat vs Dog Classifier",
+    description="Upload an image of a cat or dog and let the model classify it!"
+)
+
+# Launch the Gradio app
+interface.launch()