Updated application file

app.py

@@ -1,7 +1,19 @@
 import gradio as gr
+from fastai.vision.all import *
 
-def greet(name):
-  return f"Hello {name}! It's nice to have you on thi space."
+learn = load_learner("bird_or_forest.pkl")
+classes = ["Bird", "Forest"]
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+def classify_images(im):
+  cat, idx, probs = learn.predict(im)
+  return dict(zip(classes, map(float, probs)))
+
+image = gr.Inputs.Image(shape=(192, 192))
+label = gr.Outputs.Label()
+examples = ["forest.jpg", "bird.jpg"]
+
+iface = gr.Interface(fn=classify_images, 
+                    inputs=image, 
+                    outputs=label, 
+                    examples=examples)
 iface.launch()