Janeka commited on
Commit
a4ef453
·
verified ·
1 Parent(s): 39624be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  from rembg import remove
3
- from PIL import Image, ImageDraw
4
  import numpy as np
5
  import io
6
 
@@ -31,7 +31,7 @@ with gr.Blocks() as iface:
31
  remove_btn.click(remove_bg, inputs=input_img, outputs=output_img)
32
 
33
  gr.Markdown("### Refine Background Directly on Image")
34
- mask_editor = gr.Image(type="filepath", label="Draw on Image (Use + for Keep, - for Remove)", tool="sketch")
35
  threshold_slider = gr.Slider(0.00, 1.00, value=0.5, step=0.01, label="Threshold")
36
  refine_btn = gr.Button("Apply Refinements")
37
  refine_btn.click(refine_bg, inputs=[input_img, mask_editor, threshold_slider], outputs=output_img)
 
1
  import gradio as gr
2
  from rembg import remove
3
+ from PIL import Image
4
  import numpy as np
5
  import io
6
 
 
31
  remove_btn.click(remove_bg, inputs=input_img, outputs=output_img)
32
 
33
  gr.Markdown("### Refine Background Directly on Image")
34
+ mask_editor = gr.Image(type="filepath", label="Upload Mask (Use Black for Remove, White for Keep)")
35
  threshold_slider = gr.Slider(0.00, 1.00, value=0.5, step=0.01, label="Threshold")
36
  refine_btn = gr.Button("Apply Refinements")
37
  refine_btn.click(refine_bg, inputs=[input_img, mask_editor, threshold_slider], outputs=output_img)