Update app.py

app.py

@@ -12,10 +12,10 @@ remove_bg_models = {
 
 model_choices = list(remove_bg_models.keys())
 
-def predict(image, session, smoot, matting, bg_color):
+def predict(image, session):
     session = new_session(remove_bg_models[session])
     try:
-        return remove(session, image, smoot, matting, bg_color)
+        return remove(session, image, False, None, None)
     except ValueError as err:
         logging.error(err)
         return make_label(str(err)), None
@@ -25,13 +25,13 @@ with gr.Blocks(title="Remove background") as app:
     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")
             run_btn = gr.Button(value="Remove background", variant="primary")
             clear_btn = gr.Button(value="Clear", variant="secondary")
         with gr.Column():
             output_img = gr.Image(type="pil", label="Result image")
 
-    run_btn.click(predict, inputs=[input_img, gr.Dropdown(choices=model_choices, value="U2NET"), False, None, None],
-                  outputs=[output_img])
+    run_btn.click(predict, inputs=[input_img, drp_models], outputs=[output_img])
     clear_btn.click(lambda: (None, None), inputs=None, outputs=[input_img, output_img])
 
 app.launch(share=False, debug=True, enable_queue=True, show_error=True)