Hugging Face's logo

Spaces:
sarath2803
/
first_model
Sleeping

App Files Community
sarath2803 commited on
Commit
3f46f28
·
verified ·
1 Parent(s): 117d17c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ learn=load_learner('model.pkl')
17
  categories=('Dog','Cat')
18
 
19
  def classify_images(img):
20
- pred.idx.probs=learn.predict(img)
21
  return dict(zip(categories,map(float,probs)))
22
 
23
  image=gr.Image()
 
17
  categories=('Dog','Cat')
18
 
19
  def classify_images(img):
20
+ pred,idx,probs=learn.predict(img)
21
  return dict(zip(categories,map(float,probs)))
22
 
23
  image=gr.Image()