Update app with an image classifier

app.py

@@ -1,9 +1,18 @@
 import gradio as gr
+from fastai.vision.learner import load_learner
 
+model = load_learner("model.pkl")
 
-def greet(name):
-    return "Hello " + name + "!!"
 
+def classify_image(image):
+    preds = model.predict(image)[2]
+    return dict(zip(["Bird", "Forest"], preds.tolist()))
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+gr.Interface(
+    fn=classify_image,
+    inputs=gr.Image(),
+    outputs=gr.Label(num_top_classes=2),
+    examples=["images/bird.jpeg", "images/forest.jpeg"],
+    title="Bird or Forest?",
+).launch()