Update app.py

app.py

@@ -2,23 +2,16 @@
 # coding: utf-8
 # Dogs vs Cats
 
-# In[7]:
-
 #|export
 from fastai.vision.all import *
 import gradio as gr
+import os
 
 def is_cat(x): return x[0].isupper()
 
-
-# In[9]:
-
 #|export
 learn = load_learner('model.pkl')
 
-
-# In[12]:
-
 #|export
 categories = ('Dog', 'Cat')
 
@@ -26,11 +19,22 @@ def classify_image(img):
     pred, idx, probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
 
-
 #|export
-image = gr.components.Image(shape=(192, 192))  # Updated component syntax
-label = gr.components.Label()  # Updated component syntax
-examples = ['dog.jpg', 'cat.jpg', 'dunno.jpg']
-
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-intf.launch(inline=False, share=True)
+image = gr.components.Image(shape=(192, 192))
+label = gr.components.Label()
+examples = [
+    os.path.join(os.path.dirname(__file__), "dog.jpg"),
+    os.path.join(os.path.dirname(__file__), "cat.jpg"),
+    os.path.join(os.path.dirname(__file__), "dunno.jpg"),
+]
+
+intf = gr.Interface(
+    fn=classify_image, 
+    inputs=image, 
+    outputs=label, 
+    examples=examples,
+    flagging_options=["incorrect", "other"]  # Adjust flagging options as needed
+)
+
+if __name__ == "__main__":
+    intf.launch(inline=False, share=True)