codingwithlewis commited on
Commit
a710e8e
·
verified ·
1 Parent(s): 8af0855

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = """