Omega02gdfdd commited on
Commit
5d062c3
·
verified ·
1 Parent(s): bf91f73

Update app.py

Browse files

callback change

Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -55,15 +55,14 @@ def query_image(img, text_queries, score_threshold):
55
  result_labels.append((box, text_queries[label.item()]))
56
  result
57
  num_result_labels = len(result_labels)
58
- return img, result_labels, num_result_labels
59
-
60
 
61
 
62
 
63
  demo = gr.Interface(
64
  query_image,
65
- inputs=[gr.Image(), "text", gr.Slider(0, 1, value=0.1)],
66
- outputs=["annotatedimage","text"],
67
  title="Zero-Shot Object Detection with OWLv2",
68
  )
69
  demo.launch()
 
55
  result_labels.append((box, text_queries[label.item()]))
56
  result
57
  num_result_labels = len(result_labels)
58
+ return img, result_labels
 
59
 
60
 
61
 
62
  demo = gr.Interface(
63
  query_image,
64
+ inputs=[gr.Image(), "text", gr.Slider(0, 1, value=0.2)],
65
+ outputs=["annotatedimage"],
66
  title="Zero-Shot Object Detection with OWLv2",
67
  )
68
  demo.launch()