Spaces:
Sleeping
Sleeping
codingwithlewis
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -35,7 +35,7 @@ def query_image(img, text_queries, score_threshold):
|
|
35 |
if score < score_threshold:
|
36 |
continue
|
37 |
result_labels.append((box, text_queries[label.item()]))
|
38 |
-
return img, result_labels
|
39 |
|
40 |
|
41 |
description = """
|
|
|
35 |
if score < score_threshold:
|
36 |
continue
|
37 |
result_labels.append((box, text_queries[label.item()]))
|
38 |
+
return img, result_labels, boxes
|
39 |
|
40 |
|
41 |
description = """
|