Update Dockerfile

Dockerfile

@@ -6,10 +6,17 @@ ENV MKL_THREADING_LAYER=GNU
 # Install dependencies
 RUN apt-get update && apt-get install -y \
     git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev pkg-config \
-    build-essential cmake g++ python3-dev software-properties-common \
-    wget gnupg2 libopenblas-dev && \
+    build-essential cmake g++ python3.9 python3.9-dev python3.9-venv \
+    software-properties-common wget gnupg2 libopenblas-dev && \
     rm -rf /var/lib/apt/lists/*
 
+# Ensure Python 3.9 is the default version
+RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 && \
+    update-alternatives --set python3 /usr/bin/python3.9
+
+# Install pip for Python 3.9
+RUN curl -sS https://bootstrap.pypa.io/get-pip.py | python3.9
+
 # Add NVIDIA package repositories
 RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb && \
     dpkg -i cuda-keyring_1.0-1_all.deb && \
@@ -51,14 +58,12 @@ RUN git clone -b dev https://github.com/fffiloni/dreamtalk $HOME/app
 RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O $HOME/app/checkpoints/denoising_network.pth
 RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O $HOME/app/checkpoints/renderer.pt
 
-# Upgrade pip & install dependencies
-RUN pip install --upgrade pip setuptools wheel
-RUN pip install --no-cache-dir urllib3 transformers yacs scipy scikit-image scikit-learn PyYAML Pillow "numpy<2" opencv-python imageio ffmpeg-python av "moviepy<2" gradio
-
-# Install CMake first
-RUN pip install cmake
+# Upgrade pip & install dependencies for Python 3.9
+RUN python3.9 -m ensurepip && python3.9 -m pip install --upgrade pip setuptools wheel
 
-RUN pip install dlib==19.24.0 -vvv
+# Install dependencies using Python 3.9
+RUN python3.9 -m pip install --no-cache-dir urllib3 transformers yacs scipy scikit-image scikit-learn \
+    PyYAML Pillow "numpy<2" opencv-python imageio ffmpeg-python av "moviepy<2" gradio cmake dlib==19.24.0 -vvv
 
 # Copy application file
 COPY app.py .
@@ -68,4 +73,4 @@ ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
 # Run the application
-CMD ["python", "app.py"]
+CMD ["python3.9", "app.py"]