solve bug

app.py

@@ -31,8 +31,10 @@ def run_rembg(img):
 
     return cropped_image
 
-def run_rembg_withmodel(img,model):
-    output = remove(img)
+def run_rembg_withmodel(img, model):
+    model_name = model
+    session = new_session(model_name)
+    output = session.remove(input, session=session)
     output_pil = Image.fromarray(output)
 
     # Remove margins 
@@ -280,7 +282,7 @@ if __name__=="__main__":
 
             btn.click(
                 fn=run_rembg_withmodel,
-                inputs=[image_input, model_name]
+                inputs=[image_input, model_name],
                 outputs=image_output,
                 api_name="imageMattingWithModel"
                 )