Hugging Face's logo

Spaces:
akgeni
/
bears
Sleeping

App Files Community
akgeni commited on
Commit
49fc6b3
•
1 Parent(s): 9861e33

resoved bugs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ learner = load_learner("export.pkl")
5
 
6
  categories = ['black bear', 'grizzly bear', 'teddy bear']
7
  def classify_image(image):
8
- pred, idx, prob = learner.predict(im)
9
  return dict(zip(categories, map(float, prob)))
10
 
11
  image = gr.inputs.Image(shape=(192, 192))
 
5
 
6
  categories = ['black bear', 'grizzly bear', 'teddy bear']
7
  def classify_image(image):
8
+ pred, idx, prob = learner.predict(image)
9
  return dict(zip(categories, map(float, prob)))
10
 
11
  image = gr.inputs.Image(shape=(192, 192))