Spaces:
Paused
Paused
Commit
·
3dda9fb
1
Parent(s):
3e21e04
Update app.py
Browse files
app.py
CHANGED
@@ -550,8 +550,8 @@ def run_anything_task(input_image, text_prompt, box_threshold, text_threshold,
|
|
550 |
|
551 |
results = []
|
552 |
|
553 |
-
|
554 |
-
for mask in masks:
|
555 |
color = np.concatenate([np.random.random(3), np.array([0.6])], axis=0)
|
556 |
# color = np.array([30/255, 144/255, 255/255, 0.6])
|
557 |
show_mask(mask.cpu().numpy(), plt.gca(), color)
|
@@ -561,7 +561,6 @@ def run_anything_task(input_image, text_prompt, box_threshold, text_threshold,
|
|
561 |
"label": boxes_with_labels[i][1],
|
562 |
"color": color
|
563 |
}
|
564 |
-
i++
|
565 |
results.append(item)
|
566 |
for box, label in boxes_with_labels:
|
567 |
show_box(box.cpu().numpy(), plt.gca(), label)
|
|
|
550 |
|
551 |
results = []
|
552 |
|
553 |
+
|
554 |
+
for i, mask in enumerate(masks):
|
555 |
color = np.concatenate([np.random.random(3), np.array([0.6])], axis=0)
|
556 |
# color = np.array([30/255, 144/255, 255/255, 0.6])
|
557 |
show_mask(mask.cpu().numpy(), plt.gca(), color)
|
|
|
561 |
"label": boxes_with_labels[i][1],
|
562 |
"color": color
|
563 |
}
|
|
|
564 |
results.append(item)
|
565 |
for box, label in boxes_with_labels:
|
566 |
show_box(box.cpu().numpy(), plt.gca(), label)
|