Update app.py
Browse files
app.py
CHANGED
@@ -118,8 +118,8 @@ with gr.Blocks(title="Remove background") as app:
|
|
118 |
|
119 |
# chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
|
120 |
|
121 |
-
|
122 |
-
|
123 |
|
124 |
with gr.Row():
|
125 |
gr.HTML(footer)
|
|
|
118 |
|
119 |
# chk_dominant.change(set_dominant_color, inputs=[chk_dominant], outputs=[color_state, pkr_color])
|
120 |
|
121 |
+
run_btn.click(predict, inputs=[input_img, drp_models, chk_smoot_mask, matting_state, color_state],
|
122 |
+
outputs=[output_img, mask_img])
|
123 |
|
124 |
with gr.Row():
|
125 |
gr.HTML(footer)
|