Updating Application File

app.py

@@ -14,12 +14,7 @@ def is_cat(x):
 learn = load_learner('model.pkl')
 
 # %% ../app.ipynb 5
-categories = ('Abyssian', 'American Bulldog', 'American Pit Bull Terrier', 'Basset Hound', 'Beagle','Bengal',
-                'Birman', 'Bombay', 'Boxer','British Shorthair', 'Chihuahua',
-            'Egyptian Mau', 'English Cocker Spaniel', 'English Setter', 'German Shorthaired', 'Great Pyrenees', 'Havanese', 'Japanese Chin', 
-            'Keeshond', 'Leonberger', 'Main Coon', 'Miniature Pinscher', 'Newfoundland', 'Persian', 'Pomeranian', 'Pug', 'Ragdoll', 'Russian Blue', 
-            'St.Bernard', 'Samyoed', 'Scottish Terrier', 'Shiba Inu', 'Siamese', 'Sphynx',  
-             'Staffordshire Bull Terrier', 'Wheaten Terrier', 'Yorkshire Terrier')
+categories = learn.dls.vocab
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)