Spaces:

Rhasan97
/

snake-recognizer

Sleeping

Rhasan97 commited on Jul 12, 2023

Commit

c9c2c7b

•

1 Parent(s): 837886b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ snake_labels = (
 model = load_learner('snake-recognizer-v0.pkl')
 def recognize_snake(photo):
-  pred, probs = model.predict(photo)
   return dict(zip(snake_labels, map(float, probs)))

 model = load_learner('snake-recognizer-v0.pkl')
 def recognize_snake(photo):
+  pred,idx, probs = model.predict(photo)
   return dict(zip(snake_labels, map(float, probs)))