transparent output

app.py

@@ -56,7 +56,11 @@ def process_image(image, prompt, threhsold, alpha_value, draw_rectangles):
     ax.axis("off")
     plt.tight_layout()
 
-    return fig, mask
+    bmask = Image.fromarray(bmask.astype(np.uint8) * 255, "L")
+    output_image = Image.new("RGBA", image.size, (0, 0, 0, 0))
+    output_image.paste(image, mask=bmask)
+
+    return fig, mask, output_image
 
 
 title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
@@ -85,7 +89,8 @@ with gr.Blocks() as demo:
 
         with gr.Column():
             output_plot = gr.Plot(label="Segmentation Result")
-            output_image = gr.Image(label="Mask")
+            output_mask = gr.Image(label="Mask")
+            output_image = gr.Image(label="Output Image")
 
     btn_process.click(
         process_image,
@@ -96,7 +101,7 @@ with gr.Blocks() as demo:
             input_slider_A,
             draw_rectangles,
         ],
-        outputs=[output_plot, output_image],
+        outputs=[output_plot, output_mask, output_image],
     )
 
     gr.Examples(