boushra-kolko commited on
Commit
30888df
1 Parent(s): 48ff942

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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', 'tiger', 'lion')
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)))