Spaces:
Runtime error
Runtime error
liuyizhang
commited on
Commit
•
dae4b5a
1
Parent(s):
5247a47
update app.py
Browse files
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")
|