princeml commited on
Commit
9a523da
·
1 Parent(s): bc88526

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def gradio_wrapper(img):
20
  global model
21
  #print(np.shape(img))
22
  results = model(img,show = True) # predict on an image
23
- return results[1]
24
  demo = gr.Interface(
25
  gradio_wrapper,
26
  #gr.Image(source="webcam", streaming=True, flip=True),
 
20
  global model
21
  #print(np.shape(img))
22
  results = model(img,show = True) # predict on an image
23
+ return results[0]
24
  demo = gr.Interface(
25
  gradio_wrapper,
26
  #gr.Image(source="webcam", streaming=True, flip=True),