Update Dockerfile

Dockerfile

@@ -5,7 +5,6 @@ WORKDIR /app
 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 HOME=/tmp
 ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
@@ -23,19 +22,11 @@ RUN apt-get update && apt-get install -y \
     libopenblas-dev \
     liblapack-dev \
     libopencv-dev \
-    libomp-dev \
     && 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
-
-# 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
+RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggingface/datasets /tmp/.u2net \
+    && chmod -R 777 /tmp/uploads /tmp/results /tmp/huggingface /tmp/.u2net
 
 # Copy requirements file and install dependencies
 COPY requirements.txt .