Update app.py

app.py

@@ -10,10 +10,10 @@ remove_bg_models = {
     "U2NET Cloth Seg": "u2net_cloth_seg"
 }
 
-model_choices = list(remove_bg_models.keys())
+default_model = "U2NET"
 
-def predict(image, session):
-    session = new_session(remove_bg_models[session])
+def predict(image):
+    session = new_session(remove_bg_models[default_model])
     try:
         return remove(session, image, False, None, None)
     except ValueError as err:
@@ -25,13 +25,12 @@ 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, drp_models], outputs=[output_img])
+    run_btn.click(predict, inputs=[input_img], 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)