shengqiangShi commited on
Commit
837d4ca
·
1 Parent(s): 6dd29b3

Application file

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -70,8 +70,7 @@ def query_image(img, text_queries, score_threshold=0.5):
70
  result_labels.append((box, label_text))
71
  result_boxes.append(box)
72
 
73
- input_boxes_for_sam = [result_boxes]
74
- sam_image = generate_image_with_sam(np.array(img_pil), input_boxes_for_sam)
75
 
76
  return sam_image,result_labels
77
 
 
70
  result_labels.append((box, label_text))
71
  result_boxes.append(box)
72
 
73
+ sam_image = generate_image_with_sam(np.array(img_pil), result_boxes)
 
74
 
75
  return sam_image,result_labels
76