Spaces:

boushra-kolko
/

cats

Sleeping

boushra-kolko commited on Sep 15, 2023

Commit

6f0d661

•

1 Parent(s): fa67118

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ __all__ = ['is_cat', 'learn', 'classify_image', 'categories', 'image', 'label',
 learn = load_learner('model.pkl')
-categories = ('cat', 'lion', 'tiger','dog','person')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))

 learn = load_learner('model.pkl')
+categories = ('cat', 'tiger', 'lion','dog','person')
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))