Update app.py

app.py

@@ -146,7 +146,7 @@ def logscale(linear):
 def linscale(linear):
     return int(math.log2(linear))
 
-def remove_bg(fl, count, mh, ms, md, lm):
+def remove_bg(fl, count, mh, ms, md, lm, b, d):
     global fl_
     fr = cv2.imread(fl).astype(np.uint8)
     
@@ -205,8 +205,8 @@ def remove_bg(fl, count, mh, ms, md, lm):
         
         cv2.imwrite(fl, fr)
 
-        b = 1
-        d = 15
+        #b = 3
+        #d = 15
         element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * b + 1, 2 * b + 1), (b, b))
         mask_e = cv2.erode(mask, element) * 255
         element = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (2 * d + 1, 2 * d + 1), (d, d))
@@ -337,7 +337,9 @@ with gr.Blocks() as demo:
                         ms = gr.Slider(minimum=0, maximum=255, step=1, value=255, label="Saturation")
                         md = gr.Slider(minimum=0, maximum=255, step=1, value=12, label="Lightness")
                         lm = gr.Radio(label="Use max diffs from", choices=["average", "median", "slider"], value="slider")
-                files_orig.upload(fn=loadf, inputs=[files_orig, r_bg, mh, ms, md, lm], outputs=[files_input, gallery_input])
+                        b_size = gr.Slider(minimum=1, maximum=255, step=2, value=3, label="Inner border")
+                        d_size = gr.Slider(minimum=1, maximum=255, step=2, value=15, label="Outer border")
+                files_orig.upload(fn=loadf, inputs=[files_orig, r_bg, mh, ms, md, lm, b_size, d_size], outputs=[files_input, gallery_input])
                 
                 with gr.Row():
                     interpolation_slider = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Interpolation Steps: ")
@@ -355,12 +357,12 @@ with gr.Blocks() as demo:
     
     gr.Examples(
         examples=[[
-            ["./examples/0.png", "./examples/1.png", "./examples/2.png", "./examples/3.png", "./examples/4.png"], False, 0, 0, 0, "slider"
+            ["./examples/0.png", "./examples/1.png", "./examples/2.png", "./examples/3.png", "./examples/4.png"], False, 0, 0, 0, "slider", 1, 1
         ], [
-            ["./examples/0_flash.jpg", "./examples/1_noflash.jpg", "./examples/2_flash.jpg", "./examples/3_noflash.jpg"], True, 180, 255, 12, "slider"
+            ["./examples/0_flash.jpg", "./examples/1_noflash.jpg", "./examples/2_flash.jpg", "./examples/3_noflash.jpg"], True, 180, 255, 12, "slider", 3, 15
         ]],
         fn=loadf,
-        inputs=[files_orig, r_bg, mh, ms, md, lm],
+        inputs=[files_orig, r_bg, mh, ms, md, lm, b_size, d_size],
         outputs=[files_input, gallery_input],
         cache_examples=True
     )