fix category order

app.py

@@ -3,11 +3,11 @@ from fastbook import *
 from fastai.vision.all import *
 
 learn = load_learner("cats.pkl")
-cat_breeds = ['persian', 'siamese', 'angora', 'sphynx']
+categories = learn.dls.vocab
 
 def classify_image(img):
   pred, idx, probs = learn.predict(img)
-  return dict(zip(cat_breeds, map(float, probs)))
+  return dict(zip(categories, map(float, probs)))
 
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()