Update Dockerfile

Dockerfile

@@ -24,15 +24,17 @@ RUN apt-get update && apt-get install -y \
     liblapack-dev \
     libopencv-dev \
     libomp-dev \
-    wget \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
 RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets /tmp/numba_cache /tmp/.u2net \
     && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface /tmp/numba_cache /tmp/.u2net
 
-# Pre-download u2net.pth to the path expected by backgroundremover
-RUN wget -O /tmp/.u2net/u2net.pth https://huggingface.co/spaces/nadermx/backgroundremover/resolve/main/u2net.pth \
+# Install huggingface_hub early to download u2net.pth
+RUN pip install --no-cache-dir huggingface_hub==0.23.0
+
+# Download u2net.pth using hf_hub_download with HF_TOKEN
+RUN python -c "from huggingface_hub import hf_hub_download; hf_hub_download(repo_id='nadermx/backgroundremover', filename='u2net.pth', local_dir='/tmp/.u2net', token='${HF_TOKEN}')" \
     && chmod 644 /tmp/.u2net/u2net.pth
 
 # Copy requirements file and install dependencies