Update app.py

app.py

@@ -549,7 +549,7 @@ def apply_mask(d, b):
     cv2.imwrite(masks[frame_selected], mask) #d["background"]
     return masks[frame_selected], depths, frames
 
-def draw_mask(l, t, th, v, d, evt: gr.EventData):
+def draw_mask(l, t, b, v, d, evt: gr.EventData):
     global frames
     global depths
     global params
@@ -565,7 +565,7 @@ def draw_mask(l, t, th, v, d, evt: gr.EventData):
 
     if len(edge) == 0 or params["fnum"] != frame_selected or params["l"] != l:
       clrs = cv2.cvtColor(cv2.imread(frames[frame_selected]).astype(np.uint8), cv2.COLOR_BGR2HSV)
-      clrs = cv2.medianBlur(clrs, th)
+      clrs = cv2.medianBlur(clrs, b)
       if len(edge) > 0:
           d["background"] = cv2.imread(depths[frame_selected]).astype(np.uint8)
           
@@ -978,9 +978,8 @@ with gr.Blocks(css=css, js=js) as demo:
               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=15, maximum=256, minimum=1, step=2)
-              threshold = gr.Slider(label="Threshold", value=12, maximum=128, minimum=0, step=1)
               mouse = gr.Textbox(elem_id="mouse", value="""[]""", interactive=False)
-              mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, tolerance, threshold, mouse, output_mask], outputs=[output_mask])
+              mouse.input(fn=draw_mask, show_progress="minimal", inputs=[levels, tolerance, bsize, mouse, output_mask], outputs=[output_mask])
               apply.click(fn=apply_mask, inputs=[output_mask, bsize], outputs=[output_mask, output_depth, output_frame])
               reset.click(fn=reset_mask, inputs=None, outputs=[output_mask, output_depth])