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

Dockerfile

@@ -5,7 +5,8 @@ FROM python:3.10-slim
 ENV PYTHONDONTWRITEBYTECODE=1 \
     PYTHONUNBUFFERED=1 \
     PIP_NO_CACHE_DIR=1 \
-    HF_HOME=/root/.cache/huggingface
+    HF_HOME=/app/.cache/huggingface \
+    HF_TOKEN=""
 
 # System deps
 RUN apt-get update && apt-get install -y --no-install-recommends \
@@ -14,6 +15,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
         libgl1 \
         && rm -rf /var/lib/apt/lists/*
 
+# Ensure HF cache dir exists and is writable
+RUN mkdir -p /app/.cache/huggingface && chmod -R 777 /app/.cache
+
 WORKDIR /app
 
 # Copy only requirements first for better caching

server.py

@@ -82,7 +82,8 @@ 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")
-    net = BriaRMBG.from_pretrained("briaai/RMBG-1.4")
+    # Pass empty token to avoid reading a potentially unreadable token file in HF_HOME
+    net = BriaRMBG.from_pretrained("briaai/RMBG-1.4", token="")
     net.to(_device)
     net.eval()