Spaces:
Running
Running
File size: 527 Bytes
fb15271 763fff9 fb15271 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import os, requests
from io import BytesIO
from PIL import Image
from dotenv import load_dotenv
load_dotenv()
API_KEY = os.getenv("REMOVE_BG_API_KEY")
ENDPOINT = "https://api.remove.bg/v1.0/removebg"
def remove_background(image_bytes: bytes) -> Image.Image:
resp = requests.post(
ENDPOINT,
files={"image_file": ("image.jpg", image_bytes, "image/jpeg")},
data={"size": "auto"},
headers={"X-Api-Key": API_KEY},
)
resp.raise_for_status()
return Image.open(BytesIO(resp.content))
|