Update app.py

app.py

@@ -1,29 +1,16 @@
-import gradio as gr
-import os
-import subprocess
-
-subprocess.call(["pip", "install", "fastai"])
-
 from fastai.vision.all import *
+import gradio as gr
 
+learn = load_learner('model.pkl')
 
-def image_mod(image):
-    learn_inf = load_learner("model.pkl")
-    return learn_inf.predict(image)[0]
-
+breeds = tuple(learn.dls.vocab)
+def classify_image(image):
+  pred,idx,probs = learn.predict(image)
+  return dict(zip(breeds,map(float,probs)))
 
-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"),
-    ],
-)
+image = gr.inputs.Image(shape=(192,192))
+label = gr.outputs.Label()
+exemples = ['chihuahua.jpg','german_shepard.jpg']
 
-if __name__ == "__main__":
-    demo.launch()
+intf = gr.Interface(fn = classify_image,inputs=image,outputs=label,exemples=exemples)
+intf.launch(inline=False)