update

Dockerfile

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