update

Dockerfile

@@ -1,15 +1,15 @@
 FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
 
 RUN apt-get update && apt-get install -y \
-    python3 python3-pip python3-dev cmake build-essential \
+    git cmake build-essential python3 python3-pip python3-dev \
     libboost-all-dev libopenblas-dev liblapack-dev
 
-RUN git clone https://github.com/davisking/dlib.git /dlib \
-    && cd /dlib \
-    && mkdir build && cd build \
-    && cmake .. -DDLIB_USE_CUDA=1 \
-    && cmake --build . --config Release \
-    && python3 setup.py install
+WORKDIR /dlib
+RUN git clone https://github.com/davisking/dlib.git . \
+  && mkdir build && cd build \
+  && cmake .. -DDLIB_USE_CUDA=1 \
+  && cmake --build . --config Release
+
 
 # Choose a writable directory, e.g., /home/user/huggingface
 RUN mkdir -p /home/user/huggingface
@@ -20,11 +20,16 @@ ENV HF_HOME=/home/user/huggingface \
     HF_HUB_CACHE=/home/user/huggingface/hub \
     TRANSFORMERS_CACHE=/home/user/huggingface/hub
 
+# install via pip so dependencies like wheel are used
+RUN pip install --upgrade pip wheel cmake setuptools && \
+    cd /dlib && python3 setup.py install
+
 WORKDIR /app
-COPY requirements.txt ./
-RUN pip install --upgrade pip && pip install -r requirements.txt
+COPY requirements.txt .
+RUN pip install -r requirements.txt
 COPY . .
-CMD ["python", "app.py"]
+CMD ["python3", "app.py"]
+
 
 # FROM yogi0421/dlib-base-image:python38-dlib19.24.0