Update app.py

app.py

@@ -274,7 +274,7 @@ def sortFiles(e):
     e = e.split('/')
     return e[len(e)-1]
 
-def loadf(f, r_bg, mh, ms, md, lm):
+def loadf(f, r_bg, mh, ms, md, lm, b, d):
     if f != None and f[0] != None:
         f.sort(key=sortFiles)
         fnew = []
@@ -287,7 +287,7 @@ def loadf(f, r_bg, mh, ms, md, lm):
                 fl = denoise(fl)
 
             if r_bg == True:
-                fl = remove_bg(fl, i, mh, ms, md, lm)
+                fl = remove_bg(fl, i, mh, ms, md, lm, b, d)
                 if i % 2: # odd: is photo without the flash
                     fnew.append(fl)
             else:
@@ -337,8 +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")
-                        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")
+                        with gr.Tab("Border"):
+                            b_size = gr.Slider(minimum=1, maximum=255, step=2, value=3, label="Inner")
+                            d_size = gr.Slider(minimum=1, maximum=255, step=2, value=15, label="Outer")
                 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():