skytnt commited on
Commit
0ee8653
1 Parent(s): 7615c22

png format

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -46,7 +46,7 @@ if __name__ == "__main__":
46
  examples = gr.Examples(examples=examples_data, inputs=[input_img])
47
  run_btn = gr.Button(variant="primary")
48
  with gr.Row():
49
- output_mask = gr.Image(label="mask")
50
- output_img = gr.Image(label="result", image_mode="RGBA")
51
  run_btn.click(rmbg_fn, [input_img], [output_mask, output_img])
52
  app.launch()
 
46
  examples = gr.Examples(examples=examples_data, inputs=[input_img])
47
  run_btn = gr.Button(variant="primary")
48
  with gr.Row():
49
+ output_mask = gr.Image(label="mask", format="png")
50
+ output_img = gr.Image(label="result", image_mode="RGBA", format="png")
51
  run_btn.click(rmbg_fn, [input_img], [output_mask, output_img])
52
  app.launch()