إضافة ملفات جديدة

Dockerfile

@@ -29,6 +29,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
 
 # Copy application code
 COPY . /app
+# Optional: allow passing HF_TOKEN at runtime for private models
+# ENV HF_TOKEN=""
+
 
 # Expose port
 EXPOSE 7860

server.py

@@ -1,4 +1,5 @@
 import io
+import os
 import base64
 from typing import Optional
 
@@ -82,8 +83,9 @@ def remove_background(pil_image: Image.Image) -> Image.Image:
 async def _load_model():
     global net, _device
     _device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    # Pass empty token to avoid reading a potentially unreadable token file in HF_HOME
-    net = BriaRMBG.from_pretrained("briaai/RMBG-1.4", token="")
+    # If HF_TOKEN env is provided, use it; otherwise pass token=None to avoid sending bad headers
+    hf_token = os.environ.get("HF_TOKEN") or None
+    net = BriaRMBG.from_pretrained("briaai/RMBG-1.4", token=hf_token)
     net.to(_device)
     net.eval()