Deploy from GitHub Actions: 384de8fb9abdf39acb6709c3d26936ede947d83d

Dockerfile

@@ -3,7 +3,6 @@ FROM python:3.13-slim
 ENV PORT=7860
 
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    libgl1 \
     libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
@@ -21,10 +20,12 @@ COPY --chown=user requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
 
-RUN python -c "from rembg import new_session; new_session('u2net')"
+COPY --chown=user src/download_model.py src/download_model.py
+
+RUN python src/download_model.py
 
 COPY --chown=user . $HOME/app
 
 EXPOSE ${PORT}
 
-CMD ["/bin/sh", "-c", "exec uvicorn main:app --host 0.0.0.0 --port $PORT"]
+CMD ["/bin/sh", "-c", "exec uvicorn src.main:app --host 0.0.0.0 --port $PORT"]

package.json

@@ -3,8 +3,8 @@
   "version": "0.0.1",
   "private": true,
   "scripts": {
-    "dev": "venv\\Scripts\\activate && uvicorn main:app --reload --port 8000",
-    "start": "source venv/bin/activate && uvicorn main:app --port 8000",
+    "dev": "venv\\Scripts\\activate && uvicorn src.main:app --reload --port 8000",
+    "start": "source venv/bin/activate && uvicorn src.main:app --port 8000",
     "setup": "pip install -r requirements.txt"
   }
 }

src/download_model.py

@@ -0,0 +1,4 @@
+from rembg import new_session
+
+if __name__ == "__main__":
+    new_session("birefnet-general")

main.py → src/main.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, UploadFile, File, Response, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
-from rembg import remove
+from rembg import remove, new_session
 from PIL import Image, UnidentifiedImageError
 import io
 import os
@@ -11,6 +11,9 @@ logger = logging.getLogger(__name__)
 
 app = FastAPI(title="NoBG API")
 
+model_name = "birefnet-general"
+session = new_session(model_name)
+
 origins = os.getenv("ALLOWED_ORIGINS", "*").split(",")
 
 app.add_middleware(
@@ -32,7 +35,7 @@ def remove_background(file: UploadFile = File(...)):
 
     try:
         input_image = Image.open(file.file)
-        output_image = remove(input_image)
+        output_image = remove(input_image, session=session)
         img_byte_arr = io.BytesIO()
         output_image.save(img_byte_arr, format="PNG")
         img_byte_arr.seek(0)