Update app.py

app.py

@@ -199,7 +199,7 @@ def remove_bg(fl, count, mh, ms, md, lm):
         #fr[fr_fg==0] = (255,255,255)
 
         fr_fg[fr_fg>0] = 3 #probable fg
-        mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,31, cv2.GC_INIT_WITH_MASK)
+        mask, bgdModel, fgdModel = cv2.grabCut(fr, fr_fg, None,None,None,65, cv2.GC_INIT_WITH_MASK)
         mask = np.where((mask==2)|(mask==0),0,1).astype('uint8')
         #fr[mask==0] = (255,255,255)
         
@@ -335,7 +335,7 @@ with gr.Blocks() as demo:
                     with gr.Accordion(label="Max differences for background", open=False):
                         mh = gr.Slider(minimum=0, maximum=180, step=1, value=180, label="Hue")
                         ms = gr.Slider(minimum=0, maximum=255, step=1, value=255, label="Saturation")
-                        md = gr.Slider(minimum=0, maximum=255, step=1, value=16, label="Lightness")
+                        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])
                 
@@ -357,7 +357,7 @@ with gr.Blocks() as demo:
         examples=[[
             ["./examples/0.png", "./examples/1.png", "./examples/2.png", "./examples/3.png", "./examples/4.png"], False, 0, 0, 0, "slider"
         ], [
-            ["./examples/0_flash.jpg", "./examples/1_noflash.jpg", "./examples/2_flash.jpg", "./examples/3_noflash.jpg"], True, 180, 255, 16, "slider"
+            ["./examples/0_flash.jpg", "./examples/1_noflash.jpg", "./examples/2_flash.jpg", "./examples/3_noflash.jpg"], True, 180, 255, 12, "slider"
         ]],
         fn=loadf,
         inputs=[files_orig, r_bg, mh, ms, md, lm],