Update app.py

app.py

@@ -5,11 +5,11 @@ def which_player(x): return x[0].isupper()
 
 learn = load_learner('model.pkl')
 
-categories = ('Kevin Durant', 'Kyrie Irving', 'Ben Simmons', 'Kawhi Leonard', 'Zach Lavine')
+categories = learn.dls.vocab
 
 def classify_image(img):
     pred, ids, probs = learn.predict(img)
-    return dict(zip(pred, map(float, probs)))
+    return dict(zip(categories, map(float, probs)))
 
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()