Spaces:
Runtime error
Runtime error
Updating Application File
Browse files
app.py
CHANGED
@@ -14,12 +14,7 @@ def is_cat(x):
|
|
14 |
learn = load_learner('model.pkl')
|
15 |
|
16 |
# %% ../app.ipynb 5
|
17 |
-
categories =
|
18 |
-
'Birman', 'Bombay', 'Boxer','British Shorthair', 'Chihuahua',
|
19 |
-
'Egyptian Mau', 'English Cocker Spaniel', 'English Setter', 'German Shorthaired', 'Great Pyrenees', 'Havanese', 'Japanese Chin',
|
20 |
-
'Keeshond', 'Leonberger', 'Main Coon', 'Miniature Pinscher', 'Newfoundland', 'Persian', 'Pomeranian', 'Pug', 'Ragdoll', 'Russian Blue',
|
21 |
-
'St.Bernard', 'Samyoed', 'Scottish Terrier', 'Shiba Inu', 'Siamese', 'Sphynx',
|
22 |
-
'Staffordshire Bull Terrier', 'Wheaten Terrier', 'Yorkshire Terrier')
|
23 |
|
24 |
def classify_image(img):
|
25 |
pred, idx, probs = learn.predict(img)
|
|
|
14 |
learn = load_learner('model.pkl')
|
15 |
|
16 |
# %% ../app.ipynb 5
|
17 |
+
categories = learn.dls.vocab
|
|
|
|
|
|
|
|
|
|
|
18 |
|
19 |
def classify_image(img):
|
20 |
pred, idx, probs = learn.predict(img)
|