Rhasan97 commited on
Commit
693dc5c
1 Parent(s): c52f07e
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -26,8 +26,8 @@ model = load_learner('snake-recognizer-v0.pkl')
26
 
27
  def recognize_snake(photo):
28
  pred,idx, probs = model.predict(photo)
29
- results={class_label: pred for class_label, pred in zip(snake_labels, probs, pred)}
30
- return results
31
 
32
 
33
 
 
26
 
27
  def recognize_snake(photo):
28
  pred,idx, probs = model.predict(photo)
29
+ return dict(zip(snake_labels,map(pred),map(float, probs)))
30
+
31
 
32
 
33