Spaces:

Rhasan97
/

snake-recognizer

Sleeping

Rhasan97 commited on Sep 2, 2023

Commit

c52f07e

•

1 Parent(s): c506cae

jgdvd

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ model = load_learner('snake-recognizer-v0.pkl')
 def recognize_snake(photo):
     pred,idx, probs = model.predict(photo)
-    results={class_label: float(prob) for class_label, prob in zip(snake_labels,probs)}
     return results

 def recognize_snake(photo):
     pred,idx, probs = model.predict(photo)
+    results={class_label: pred for class_label, pred in zip(snake_labels, probs, pred)}
     return results