Update app.py

app.py

@@ -75,48 +75,48 @@ with gr.Blocks(title="Remove background") as app:
     gr.HTML("<center><h1>Background Remover</h1></center>")
     with gr.Row(equal_height=False):
         with gr.Column():
-           # input_img = gr.Image(type="pil", label="Input image")
-           # drp_models = gr.Dropdown(choices=model_choices, label="Model Segment", value="U2NET")
-           # with gr.Row():
-               # chk_include_matting = gr.Checkbox(label="Matting", value=False)
-              #  chk_smoot_mask = gr.Checkbox(label="Smoot Mask", value=False)
-              #  chk_show_mask = gr.Checkbox(label="Show Mask", value=False)
-           # with gr.Box(visible=False) as slider_matting:
-              #  slr_fg_threshold = gr.Slider(0, 300, value=270, step=1, label="Alpha matting foreground threshold")
-              #  slr_bg_threshold = gr.Slider(0, 50, value=20, step=1, label="Alpha matting background threshold")
-               # slr_erode_size = gr.Slider(0, 20, value=11, step=1, label="Alpha matting erode size")
-            #with gr.Box():
-               # with gr.Row():
-                   # chk_change_color = gr.Checkbox(label="Change background color", value=False)
-                   # pkr_color = gr.ColorPicker(label="Pick a new color", visible=False)
-                    #chk_dominant = gr.Checkbox(label="Use dominant color", value=False, visible=False)
+            input_img = gr.Image(type="pil", label="Input image")
+            drp_models = gr.Dropdown(choices=model_choices, label="Model Segment", value="U2NET")
+            with gr.Row():
+                chk_include_matting = gr.Checkbox(label="Matting", value=False)
+                chk_smoot_mask = gr.Checkbox(label="Smoot Mask", value=False)
+                chk_show_mask = gr.Checkbox(label="Show Mask", value=False)
+            with gr.Box(visible=False) as slider_matting:
+                slr_fg_threshold = gr.Slider(0, 300, value=270, step=1, label="Alpha matting foreground threshold")
+                slr_bg_threshold = gr.Slider(0, 50, value=20, step=1, label="Alpha matting background threshold")
+                slr_erode_size = gr.Slider(0, 20, value=11, step=1, label="Alpha matting erode size")
+            with gr.Box():
+                with gr.Row():
+                    chk_change_color = gr.Checkbox(label="Change background color", value=False)
+                    pkr_color = gr.ColorPicker(label="Pick a new color", visible=False)
+                    chk_dominant = gr.Checkbox(label="Use dominant color", value=False, visible=False)
             run_btn = gr.Button(value="Remove background", variant="primary")
         with gr.Column():
             output_img = gr.Image(type="pil", label="Image Result")
             mask_img = gr.Image(type="pil", label="Image Mask", visible=False)
-           # gr.ClearButton(components=[input_img, output_img, mask_img])
+            gr.ClearButton(components=[input_img, output_img, mask_img])
 
-    #chk_include_matting.change(change_include_matting, inputs=[chk_include_matting],
-                              # outputs=[slider_matting, matting_state,
-                                    #    slr_fg_threshold, slr_bg_threshold, slr_erode_size])
+    chk_include_matting.change(change_include_matting, inputs=[chk_include_matting],
+                               outputs=[slider_matting, matting_state,
+                                        slr_fg_threshold, slr_bg_threshold, slr_erode_size])
 
-    #slr_fg_threshold.change(change_foreground_threshold, inputs=[slr_fg_threshold, matting_state],
-                          #  outputs=[matting_state])
+    slr_fg_threshold.change(change_foreground_threshold, inputs=[slr_fg_threshold, matting_state],
+                            outputs=[matting_state])
 
-    #slr_bg_threshold.change(change_background_threshold, inputs=[slr_bg_threshold, matting_state],
-                     #       outputs=[matting_state])
+    slr_bg_threshold.change(change_background_threshold, inputs=[slr_bg_threshold, matting_state],
+                            outputs=[matting_state])
 
-   # slr_erode_size.change(change_erode_size, inputs=[slr_erode_size, matting_state],
-                  #        outputs=[matting_state])
+    slr_erode_size.change(change_erode_size, inputs=[slr_erode_size, matting_state],
+                          outputs=[matting_state])
 
-    #chk_show_mask.change(change_show_mask, inputs=[chk_show_mask], outputs=[mask_img])
+    chk_show_mask.change(change_show_mask, inputs=[chk_show_mask], outputs=[mask_img])
 
-   # chk_change_color.change(change_background_mode, inputs=[chk_change_color],
-                         #   outputs=[pkr_color, chk_dominant])
+    chk_change_color.change(change_background_mode, inputs=[chk_change_color],
+                            outputs=[pkr_color, chk_dominant])
 
-  #  pkr_color.change(change_picker_color, inputs=[pkr_color, chk_dominant], outputs=[color_state])
+    pkr_color.change(change_picker_color, inputs=[pkr_color, chk_dominant], outputs=[color_state])
 
-  #  chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
+    chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
 
     run_btn.click(predict, inputs=[input_img, drp_models, chk_smoot_mask, matting_state, color_state],
                  outputs=[output_img, mask_img])