Update app.py

app.py

@@ -16,7 +16,7 @@ from functools import partial
 # -------------------------
 MODEL_DIR = "models/BiRefNet"
 os.makedirs(MODEL_DIR, exist_ok=True)
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cpu"  # force CPU usage
 birefnet = None  # will initialize on startup
 
 # -------------------------
@@ -26,7 +26,7 @@ birefnet = None  # will initialize on startup
 async def lifespan(app: FastAPI):
     global birefnet
     if birefnet is None:
-        print("Loading BiRefNet model...")
+        print("Loading BiRefNet model on CPU...")
         birefnet = AutoModelForImageSegmentation.from_pretrained(
             "ZhengPeng7/BiRefNet",
             cache_dir=MODEL_DIR,
@@ -72,10 +72,6 @@ def process_image(image: Image.Image) -> Image.Image:
 # -------------------------
 @app.post("/remove-background")
 async def remove_background(file: UploadFile = File(None), image_url: str = Form(None)):
-    """
-    Accept either an uploaded file or an image URL.
-    Returns PNG with transparent background.
-    """
     try:
         if file:
             image = Image.open(BytesIO(await file.read())).convert("RGB")