Update app.py

app.py

@@ -1,6 +1,31 @@
 import requests
-
 import gradio as gr
-import fastai
+from fastai.vision.all import *
+import pickle
+
+# Load the trained model
+with open("model.pkl", "rb") as f:
+    learn = pickle.load(f)
+
+# Define the predict function
+def predict(image):
+    # Load the input image
+    img = PILImage.create(image)
+
+    # Perform inference
+    predicted_class, _, _ = learn.predict(img)
+
+    return predicted_class
+
+# Create Gradio interface
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(upload_button=True, label="Upload a picture of a dog"),
+    outputs="text",
+    live=True,
+    title="Dog Breed Classifier",
+    description="Upload a picture of a dog, and the model will classify it as Labrador, Husky, or Bulldog.",
+)
 
-gr.Interface(fn=predict,inputs="image",outputs="label").launch()
+# Launch the Gradio interface
+iface.launch()