Spaces:
Sleeping
Sleeping
Commit
•
30888df
1
Parent(s):
48ff942
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ __all__ = ['is_cat', 'learn', 'classify_image', 'categories', 'image', 'label',
|
|
| 8 |
learn = load_learner('model_old.pkl')
|
| 9 |
|
| 10 |
|
| 11 |
-
categories = ('cat', '
|
| 12 |
def classify_image(img):
|
| 13 |
pred,idx,probs = learn.predict(img)
|
| 14 |
return dict(zip(categories, map(float,probs)))
|
|
|
|
| 8 |
learn = load_learner('model_old.pkl')
|
| 9 |
|
| 10 |
|
| 11 |
+
categories = ('cat', 'lion', 'tiger')
|
| 12 |
def classify_image(img):
|
| 13 |
pred,idx,probs = learn.predict(img)
|
| 14 |
return dict(zip(categories, map(float,probs)))
|