remove orig img from output
Browse files
app.py
CHANGED
@@ -59,8 +59,7 @@ def extract_image(img, pos_prompts, neg_prompts, threshold):
|
|
59 |
output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
|
60 |
output_image.paste(img, mask=final_mask)
|
61 |
|
62 |
-
|
63 |
-
return output_image, final_mask, orig_image, inverse_mask
|
64 |
|
65 |
|
66 |
title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
|
@@ -91,10 +90,8 @@ with gr.Blocks() as demo:
|
|
91 |
with gr.Column():
|
92 |
output_image = gr.Image(label="Result")
|
93 |
output_mask = gr.Image(label="Mask")
|
94 |
-
orig_image = gr.Image(label="Orig")
|
95 |
inverse_mask = gr.Image(label="Inverse")
|
96 |
|
97 |
-
|
98 |
btn_process.click(
|
99 |
extract_image,
|
100 |
inputs=[
|
@@ -103,7 +100,7 @@ with gr.Blocks() as demo:
|
|
103 |
negative_prompts,
|
104 |
input_slider_T,
|
105 |
],
|
106 |
-
outputs=[output_image, output_mask,
|
107 |
api_name="mask"
|
108 |
)
|
109 |
|
|
|
59 |
output_image = Image.new("RGBA", img.size, (0, 0, 0, 0))
|
60 |
output_image.paste(img, mask=final_mask)
|
61 |
|
62 |
+
return output_image, final_mask, inverse_mask
|
|
|
63 |
|
64 |
|
65 |
title = "Interactive demo: zero-shot image segmentation with CLIPSeg"
|
|
|
90 |
with gr.Column():
|
91 |
output_image = gr.Image(label="Result")
|
92 |
output_mask = gr.Image(label="Mask")
|
|
|
93 |
inverse_mask = gr.Image(label="Inverse")
|
94 |
|
|
|
95 |
btn_process.click(
|
96 |
extract_image,
|
97 |
inputs=[
|
|
|
100 |
negative_prompts,
|
101 |
input_slider_T,
|
102 |
],
|
103 |
+
outputs=[output_image, output_mask, inverse_mask],
|
104 |
api_name="mask"
|
105 |
)
|
106 |
|