Update Dockerfile

Dockerfile

@@ -19,19 +19,19 @@ ENV DEBIAN_FRONTEND=noninteractive \
     TORCH_CUDA_ARCH_LIST="8.6" \
     CMAKE_PREFIX_PATH="/usr/local/cuda:/usr/lib/x86_64-linux-gnu/"
 
-# Override PATH so that the system's /usr/bin (with Python 3.9) is used instead of Conda's
+# Override PATH so that system Python 3.9 is used in favor of Conda's Python 3.10
 ENV PATH="/usr/local/cuda/bin:/usr/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/sbin"
 
-# Install Python 3.9 and build dependencies (including for NumPy)
+# Install Python 3.9 and build dependencies (including those needed for NumPy)
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-    python3.9 python3.9-dev python3.9-venv \
-    git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev pkg-config \
-    build-essential gfortran libopenblas-dev liblapack-dev \
-    cmake g++ software-properties-common gnupg2 && \
+      python3.9 python3.9-dev python3.9-venv \
+      git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev pkg-config \
+      build-essential gfortran libopenblas-dev liblapack-dev \
+      cmake g++ software-properties-common gnupg2 && \
     rm -rf /var/lib/apt/lists/*
 
-# Set Python 3.9 as the default python3 and install pip for it
+# Set Python 3.9 as default for python3, install pip, and create symlinks
 RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 && \
     update-alternatives --set python3 /usr/bin/python3.9 && \
     curl -sS https://bootstrap.pypa.io/get-pip.py | python3.9 && \
@@ -46,16 +46,14 @@ RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86
     dpkg -i cuda-keyring_1.0-1_all.deb && \
     apt-get update && \
     apt-get install -y --no-install-recommends \
-    cuda-toolkit-11-7 libcudnn8 libcudnn8-dev && \
+      cuda-toolkit-11-7 libcudnn8 libcudnn8-dev && \
     rm -rf /var/lib/apt/lists/*
 
-# Create a non-root user
+# Create a non-root user and switch to it
 RUN useradd -m -u 1000 user
-
-# Switch to the new user
 USER user
 
-# Set the working directory
+# Set working directory
 WORKDIR /home/user/app
 
 # Clone the repository
@@ -66,8 +64,8 @@ RUN mkdir -p checkpoints && \
     wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O checkpoints/denoising_network.pth && \
     wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O checkpoints/renderer.pt
 
-# Install NumPy first (to help avoid build issues)
-RUN python3.9 -m pip install numpy
+# Pre-install NumPy (version as specified in requirements_HF.txt) to avoid build issues
+RUN python3.9 -m pip install numpy==1.26.4
 
 # Copy and install remaining Python dependencies
 COPY requirements_HF.txt .
@@ -77,9 +75,9 @@ RUN python3.9 -m pip install --no-cache-dir -r requirements_HF.txt && \
 # Copy the application file
 COPY app.py .
 
-# Set CUDA environment variables (if needed)
+# Set additional CUDA environment variables if needed
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID \
     CUDA_VISIBLE_DEVICES=0
 
-# Run the application with Python 3.9 explicitly
+# Run the application explicitly with Python 3.9
 CMD ["python3.9", "app.py"]