return orig image

app.py

@@ -57,7 +57,9 @@ def extract_image(img, pos_prompts, neg_prompts, threhsold):
     final_mask = Image.fromarray(final_mask.astype(np.uint8) * 255, "L")
     output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
     output_image.paste(img, mask=final_mask)
-    return output_image, final_mask
+
+    orig_image = img
+    return output_image, final_mask, orig_image
 
 
 title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
@@ -88,6 +90,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             output_image = gr.Image(label="Result")
             output_mask = gr.Image(label="Mask")
+            orig_image = gr.Image(label="Orig")
 
     btn_process.click(
         extract_image,
@@ -97,7 +100,7 @@ with gr.Blocks() as demo:
             negative_prompts,
             input_slider_T,
         ],
-        outputs=[output_image, output_mask],
+        outputs=[output_image, output_mask, orig_image],
         api_name="mask"
     )