changing labels

app.py

@@ -3,8 +3,8 @@ from fastai.vision.all import *
 
 # load model
 learn = load_learner('seasons_model.pkl')
-categories = ('Fall', 'Winter', 'Spring', 'Summer')
-# categories = learn.dls.vocab
+# categories = ('Fall', 'Winter', 'Spring', 'Summer')
+categories = learn.dls.vocab
 
 # classifier
 def classify_image(img):
@@ -18,4 +18,4 @@ examples = ['fall_photo.jpeg', 'winter_photo.jpeg', 'spring_photo.jpeg', 'summer
 
 # Interface
 iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-iface.launch(inline=False)
+iface.launch(inline=False)