Update app.py

app.py

@@ -136,7 +136,7 @@ def infer(f_in, interpolation, fps_output):
     return final_vid, files
 
 
-def remove_bg(fl):
+def remove_bg(fl, s, l, v):
     frame = cv2.imread(fl).astype(np.uint8)
     
     b = 5
@@ -165,8 +165,8 @@ def remove_bg(fl):
             blur_s[i,j] = (d/512).astype(np.uint8)
 
     #remove regions of low saturation and lightness (get scene without shadow)
-    m = cv2.inRange(cv2.cvtColor(frame_c, cv2.COLOR_RGB2HSV), np.array([0,0,0]), np.array([180,32,64]))
-    mask = cv2.inRange(blur_s, 0, 16)
+    m = cv2.inRange(cv2.cvtColor(frame_c, cv2.COLOR_RGB2HSV), np.array([0,0,0]), np.array([180,s,l]))
+    mask = cv2.inRange(blur_s, 0, v)
     masks = np.bitwise_and(m, mask)
     frame[masks>0] = (127,127,127)
     frame = cv2.medianBlur(frame, b)
@@ -292,7 +292,11 @@ with gr.Blocks() as demo:
                 with gr.Accordion(label="Upload files here", open=True):
                     files_input = gr.File(file_count="multiple", file_types=['image', '.mp4'])
                 gallery_input = gr.Gallery(label="Slideshow", preview=True, columns=8192, interactive=False)
-                files_input.upload(fn=loadf, inputs=[files_input], outputs=[files_input, gallery_input])
+                files_input.upload(fn=loadf, inputs=[files_input, max_s, max_l, max_v], outputs=[files_input, gallery_input])
+                with gr.Accordion(label="Background removal settings", open=False):
+                    max_s = gr.Slider(minimum=0, maximum=255, step=1, value=32, label="Shadow max saturation")
+                    max_l = gr.Slider(minimum=0, maximum=255, step=1, value=64, label="Shadow max lightness")
+                    max_v = gr.Slider(minimum=0, maximum=255, step=1, value=16, label="Shadow max variance")
                 with gr.Row():
                     interpolation_slider = gr.Slider(minimum=1, maximum=5, step=1, value=1, label="Interpolation Steps: ")
                     interpolation = gr.Number(value=1, show_label=False, interactive=False)
@@ -310,7 +314,7 @@ with gr.Blocks() as demo:
     gr.Examples(
         examples=[[["./examples/0.png", "./examples/1.png", "./examples/2.png", "./examples/3.png", "./examples/4.png"], 1, 0]],
         fn=loadf,
-        inputs=[files_input],
+        inputs=[files_input, max_s, max_l, max_v],
         outputs=[files_input, gallery_input],
         cache_examples=True
     )