Update app.py

app.py

@@ -3,12 +3,11 @@ __all__ = ['learn', 'classify_image', 'categories', 'image', 'label', 'intf']
 from fastai.vision.all import *
 import gradio as gr
 
-learn = load_learner('export (3).pkl')
+learn = load_learner('export (1).pkl')
 
-categories = ('Dog Destroying Stuff','Neutral')
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
-    return dict(zip(categories, map(float, probs)))
+    return pred
 
 image=gr.Image()
 label = gr.Label()