Add dog classification

app.py

@@ -1,7 +1,25 @@
 import gradio as gr
+import os
+from fastai.vision.all import *
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def image_mod(image):
+    learn_inf = load_learner("model.pkl")
+    return learn_inf.predict(image)[0]
+
+
+demo = gr.Interface(
+    image_mod,
+    gr.Image(type="pil"),
+    "image",
+    flagging_options=["blurry", "incorrect", "other"],
+    examples=[
+        os.path.join(os.path.dirname(__file__), "images/cheetah1.jpg"),
+        os.path.join(os.path.dirname(__file__), "images/lion.jpg"),
+        os.path.join(os.path.dirname(__file__), "images/logo.png"),
+        os.path.join(os.path.dirname(__file__), "images/tower.jpg"),
+    ],
+)
+
+if __name__ == "__main__":
+    demo.launch()