Spaces:

Rhasan97
/

snake-recognizer

Sleeping

Rhasan97 commited on Jul 12, 2023

Commit

3b5310d

•

1 Parent(s): 654ceda

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,8 +25,8 @@ snake_labels = (
 model = load_learner('snake-recognizer-v0.pkl')
 def recognize_snake(photo):
-  pred,idx, probs = model.predict(photo)
-      return dict(zip(pred, map(float,probs))
 image = gr.inputs.Image(shape=(192,192))

 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))
 image = gr.inputs.Image(shape=(192,192))