Update Dockerfile

Dockerfile

@@ -9,6 +9,8 @@ ENV PYTHONDONTWRITEBYTECODE=1
 ENV PYTHONUNBUFFERED=1
 ENV CUDA_VISIBLE_DEVICES=""
 
+ENV HF_TOKEN=${HF_TOKEN}
+
 RUN apt-get update && apt-get install -y \
     build-essential \
     libgl1-mesa-glx \
@@ -19,8 +21,8 @@ RUN apt-get update && apt-get install -y \
     cmake \
     libopenblas-dev \
     liblapack-dev \
-    git \
     libopencv-dev \
+    libomp-dev \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
@@ -31,6 +33,7 @@ RUN mkdir -p /tmp/uploads /tmp/results /tmp/huggingface/transformers /tmp/huggin
 COPY requirements.txt .
 RUN pip install --no-cache-dir -U pip && \
     pip install --no-cache-dir -r requirements.txt
+    pip cache purge