shengqiangShi commited on
Commit
6e764cd
·
1 Parent(s): b5fde6d

Application file

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