masked image now has color with a transparent background

app.py

@@ -775,13 +775,13 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
         # PROCESSING MASKS AND LABELS AND SEGMENTS
         for i, mask in enumerate(masks):
             show_mask(mask.cpu().numpy(), plt.gca(), random_color=True)
+            np_mask = mask.cpu().numpy()[0]
+            np_mask = np.expand_dims(np_mask, axis=-1)
 
-            np_mask = mask.cpu().numpy().astype(np.uint8)
-            color = np.array([255, 255, 255])
-            h, w  = np_mask.shape[-2:]
-            mask_input = np_mask.reshape(h, w, 1) * color.reshape(1, 1, -1)
+            masked_array = np.where(np_mask, np.array(image_pil.convert("RGBA")), [0, 0, 0, 0])
+            masked_image = Image.fromarray(masked_array.astype('uint8'))
+            mask_and_phrase.append((masked_image, pred_phrases[i]))
 
-            mask_and_phrase.append((mask_input, pred_phrases[i]))
         for box, label in zip(boxes_filt, pred_phrases):
             phrase_to_box[label] = list(box.cpu().numpy())
             show_box(box.cpu().numpy(), plt.gca(), label)