sessex commited on
Commit
6fc9e40
1 Parent(s): 2c4258f

remove orig img from output

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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
- orig_image = img
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, orig_image, inverse_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