hank1229 commited on
Commit
763fff9
·
verified ·
1 Parent(s): 0b94cd5

Update utils/bg_removal.py

Browse files
Files changed (1) hide show
  1. 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 ={"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))
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))