Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
+from fastai.vision.all import *
 
-def image_classifier(inp):
-    return {'cat': 0.3, 'dog': 0.7}
+def classifier(img):
+    categories = ('Gato','Perro')
+    pred,indexs,probs = learner.predict(img)
+    return dict(zip(categories,map(float,probs)))
 
-demo = gr.Interface(fn=image_classifier, inputs="image", outputs="label")
+learner = load_learner('cat.pkl')
+demo = gr.Interface(fn=classifier, inputs="image", outputs="label")
 demo.launch()