Rhasan97 commited on
Commit
837886b
1 Parent(s): 7b951b3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ snake_labels = (
25
  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, map(float, probs)))
30
 
31
 
@@ -42,4 +42,4 @@ examples = [
42
  ]
43
 
44
  iface = gr.Interface(fn=recognize_snake, inputs=image, outputs=label, examples=examples)
45
- iface.launch(inline=False)
 
25
  model = load_learner('snake-recognizer-v0.pkl')
26
 
27
  def recognize_snake(photo):
28
+ pred, probs = model.predict(photo)
29
  return dict(zip(snake_labels, map(float, probs)))
30
 
31
 
 
42
  ]
43
 
44
  iface = gr.Interface(fn=recognize_snake, inputs=image, outputs=label, examples=examples)
45
+ iface.launch(inline=False,share=True)