Update app.py

app.py

@@ -434,7 +434,8 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
       #for i in range(l):
       #    bg[(bg >= i*255/l) & (bg < (i+1)*255/l)] = i*255/(l-1)
       bg = cv2.convertScaleAbs(palette[bg]).astype(np.uint8) # Converting image back to uint
-      bg[(mask>0) | (bg==255)] = 254
+      bg[mask>0] = 254
+      bg[bg==255] = 254
 
       params["fnum"] = frame_selected
       params["l"] = l
@@ -568,7 +569,7 @@ with gr.Blocks(css=css, js=js) as demo:
             with gr.Accordion(label="Edge", open=False):
               levels = gr.Slider(label="Color levels", value=16, maximum=32, minimum=2, step=1)
               tolerance = gr.Slider(label="Tolerance", value=1, maximum=7, minimum=0, step=1)
-              bsize = gr.Slider(label="Border size", value=1, maximum=256, minimum=1, step=2)
+              bsize = gr.Slider(label="Border size", value=15, maximum=256, minimum=1, step=2)
               mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
               mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, tolerance, mouse, output_mask], outputs=[output_mask])
               output_mask.apply(fn=apply_mask, inputs=[output_mask, bsize], outputs=[output_mask, depth_file])