fix

app.py

@@ -2,18 +2,17 @@ import gradio as gr
 from fastai.vision.all import *
 import json
 
-def is_cat(x): return x[0].isupper() 
-
-
 learn = load_learner('model.pkl')
 
 with open('categories.json', 'r') as f:
-    labels = json.load(f)
+    categories = json.load(f)
+
+labels = learn.dls.vocab
 
 def predict(img):
     img = PILImage.create(img)
-    pred,pred_idx,probs = learn.predict(img)
-    return {labels[str(i)]: float(probs[i]) for i in range(len(labels))}
+    _, _, probs = learn.predict(img)
+    return {f"{categories[labels[i]]} ({labels[i]})": float(probs[i]) for i in range(len(labels))}
 
 
 title = "Predict flower species"