Spaces:
Sleeping
Sleeping
Update app.py
Browse filescallback change
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
|
59 |
-
|
60 |
|
61 |
|
62 |
|
63 |
demo = gr.Interface(
|
64 |
query_image,
|
65 |
-
inputs=[gr.Image(), "text", gr.Slider(0, 1, value=0.
|
66 |
-
outputs=["annotatedimage"
|
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()
|