liuyizhang commited on
Commit
dae4b5a
1 Parent(s): 5247a47

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -576,7 +576,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
576
  for mask in masks:
577
  show_mask(mask.cpu().numpy(), plt.gca(), random_color=True)
578
  for box, label in zip(boxes_filt, pred_phrases):
579
- show_box(box.numpy(), plt.gca(), label)
580
  plt.axis('off')
581
  image_path = os.path.join(output_dir, f"grounding_seg_output_{file_temp}.jpg")
582
  plt.savefig(image_path, bbox_inches="tight")
 
576
  for mask in masks:
577
  show_mask(mask.cpu().numpy(), plt.gca(), random_color=True)
578
  for box, label in zip(boxes_filt, pred_phrases):
579
+ show_box(box.cpu().numpy(), plt.gca(), label)
580
  plt.axis('off')
581
  image_path = os.path.join(output_dir, f"grounding_seg_output_{file_temp}.jpg")
582
  plt.savefig(image_path, bbox_inches="tight")