inverse mask option

app.py

@@ -52,14 +52,16 @@ def extract_image(img, pos_prompts, neg_prompts, threshold):
     pos_mask = np.any(np.stack(positive_masks), axis=0)
     neg_mask = np.any(np.stack(negative_masks), axis=0)
     final_mask = pos_mask & ~neg_mask
+    inverse_mask = pos_mask
 
     # extract the final image
     final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
+    inverse_mask = Image.fromarray(pos_mask.astype(np.uint8) * 255, "L")
     output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
     output_image.paste(img, mask=final_mask)
 
     orig_image = img
-    return output_image, final_mask, orig_image
+    return output_image, final_mask, orig_image, inverse_mask
 
 
 title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
@@ -91,6 +93,8 @@ with gr.Blocks() as demo:
             output_image = gr.Image(label="Result")
             output_mask = gr.Image(label="Mask")
             orig_image = gr.Image(label="Orig")
+            inverse_mask = gr.Image(label="Inverse")
+
 
     btn_process.click(
         extract_image,
@@ -100,7 +104,7 @@ with gr.Blocks() as demo:
             negative_prompts,
             input_slider_T,
         ],
-        outputs=[output_image, output_mask, orig_image],
+        outputs=[output_image, output_mask, orig_image, inverse_mask],
         api_name="mask"
     )