arham061 commited on
Commit
c9cf370
1 Parent(s): 73a47d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ learn = load_learner('model.pkl')
9
 
10
  categories = ('Dog', 'Cat')
11
 
12
- def classify_images(img):
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15
 
 
9
 
10
  categories = ('Dog', 'Cat')
11
 
12
+ def classify_image(img):
13
  pred, idx, probs = learn.predict(img)
14
  return dict(zip(categories, map(float, probs)))
15