add categories

app.py

@@ -4,10 +4,10 @@ import gradio as gr
 def is_cat(x): return x[0].isupper() 
 
 learn = load_learner('model.pkl')
-
+categories = ('Dog', 'Cat')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
-    return dict(zip(categories, map(float, rpobs)))
+    return dict(zip(categories, map(float, probs)))
 
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()