Update app.py

app.py

@@ -16,9 +16,9 @@ LABELS = LEARN.dls.vocab
 
 def gradio_predict(img):
     img = img.resize((512, 512))
-    img = PILImage.create(img)
+    # img = PILImage.create(img)
     _pred, _pred_idx, probs = LEARN.predict(img)
-    labels_probs = dict(zip(categories, map(float,probs)))
+    labels_probs = dict(zip(LABELS, map(float,probs)))
     return labels_probs
 
 # with open('gradio_article.md') as f: