Spaces:

AbdoIR
/

background-remover

Running

AbdoIR commited on Jun 22, 2025

Commit

f0e753b

verified ·

1 Parent(s): 3b8a201

Update api.py

Files changed (1) hide show

api.py CHANGED Viewed

@@ -13,7 +13,9 @@ def index():
 def remove_bg():
     file = request.files['image']
     input_image = Image.open(file)
-    output = remove(input_image)
     buffer = io.BytesIO()
     output.save(buffer, format='PNG')

 def remove_bg():
     file = request.files['image']
     input_image = Image.open(file)
+    session = new_session("u2net", providers=["CPUExecutionProvider"])
+    output = remove(input_image, session=session)
     buffer = io.BytesIO()
     output.save(buffer, format='PNG')