Spaces:
Runtime error
Runtime error
Updating app
Browse files
app.py
CHANGED
@@ -14,12 +14,12 @@ def is_cat(x):
|
|
14 |
learn = load_learner('model.pkl')
|
15 |
|
16 |
# %% ../app.ipynb 5
|
17 |
-
categories = ('American Bulldog', 'American Pit Bull Terrier', 'Basset Hound',
|
18 |
-
|
19 |
-
'
|
20 |
-
'
|
21 |
-
'
|
22 |
-
|
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 = ('Abyssian', 'American Bulldog', 'American Pit Bull Terrier', 'Basset Hound', 'Beagle','Bengal',
|
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)
|