Hugging Face's logo

Spaces:
Rhasan97
/
snake-recognizer
Sleeping

App Files Community
Rhasan97 commited on
Commit
654ceda
•
1 Parent(s): 300d06b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ model = load_learner('snake-recognizer-v0.pkl')
26
 
27
  def recognize_snake(photo):
28
  pred,idx, probs = model.predict(photo)
29
- return dict(zip(snake_labels, pred))
30
 
31
 
32
  image = gr.inputs.Image(shape=(192,192))
 
26
 
27
  def recognize_snake(photo):
28
  pred,idx, probs = model.predict(photo)
29
+ return dict(zip(pred, map(float,probs))
30
 
31
 
32
  image = gr.inputs.Image(shape=(192,192))