Update Dockerfile

Dockerfile

@@ -6,11 +6,10 @@ ENV HF_HOME=/tmp/huggingface
 ENV TRANSFORMERS_CACHE=/tmp/huggingface/transformers
 ENV HF_DATASETS_CACHE=/tmp/huggingface/datasets
 ENV NUMBA_CACHE_DIR=/tmp/numba_cache
-ENV U2NET_HOME=/tmp/u2net
+ENV HOME=/tmp
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV CUDA_VISIBLE_DEVICES=""
-
 ENV HF_TOKEN=${HF_TOKEN}
 
 RUN apt-get update && apt-get install -y \
@@ -25,25 +24,25 @@ 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
+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 \
+    && chmod 644 /tmp/.u2net/u2net.pth
 
 # Copy requirements file and install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -U pip && \
     pip install --no-cache-dir -r requirements.txt && \
     pip cache purge
-    
-
-
 
 COPY app.py .
 
-
-
 EXPOSE 7860
 
 CMD ["python", "app.py"]