Update app.py

app.py

@@ -1,25 +1,26 @@
-import gradio as gr
-import os
-from fastai.vision.all import *
-
-
-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__":
+!pip install fastai
+import gradio as gr
+import os
+from fastai.vision.all import *
+
+
+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()