Update app.py

app.py

@@ -20,6 +20,10 @@ for dirname, _, filenames in os.walk('/kaggle/input'):
 
 import gradio as gr
 
+def classify_image(img):
+    pred,idx,probs = learn.predict(img)
+    return dict(zip(categories, map(float,probs)))
+
 # Assuming no shape needs to be specified directly in the constructor
 image = gr.Image()
 label = gr.Label()