added explicit categories before classify_bats call

app.py

@@ -3,9 +3,11 @@ import gradio as gr
 
 learn = load_learner('model.pkl')
 
+categories = ('!bat', 'bat')
+
 def classify_bat(img):
-    label,_,probs = learn.predict(img)
-    return dict(zip(label, map(float, probs)))
+    label,_,probs = learn.predict(img) 
+    return dict(zip(categories, map(float, probs)))
 
 image = gr.inputs.Image(shape=(200, 200))
 label = gr.outputs.Label()