Spaces:
Running
Running
Update utils/bg_removal.py
Browse files- utils/bg_removal.py +3 -4
utils/bg_removal.py
CHANGED
@@ -11,10 +11,9 @@ ENDPOINT = "https://api.remove.bg/v1.0/removebg"
|
|
11 |
def remove_background(image_bytes: bytes) -> Image.Image:
|
12 |
resp = requests.post(
|
13 |
ENDPOINT,
|
14 |
-
files
|
15 |
-
data
|
16 |
-
headers
|
17 |
)
|
18 |
resp.raise_for_status()
|
19 |
return Image.open(BytesIO(resp.content))
|
20 |
-
|
|
|
11 |
def remove_background(image_bytes: bytes) -> Image.Image:
|
12 |
resp = requests.post(
|
13 |
ENDPOINT,
|
14 |
+
files={"image_file": ("image.jpg", image_bytes, "image/jpeg")},
|
15 |
+
data={"size": "auto"},
|
16 |
+
headers={"X-Api-Key": API_KEY},
|
17 |
)
|
18 |
resp.raise_for_status()
|
19 |
return Image.open(BytesIO(resp.content))
|
|