Update Dockerfile

Dockerfile

@@ -1,44 +1,53 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+# Use a newer PyTorch image with Python 3.11
+FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Set the MKL_THREADING_LAYER environment variable to GNU
-ENV MKL_THREADING_LAYER=GNU
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential cmake \
+    libboost-all-dev python3-dev
 
-RUN apt-get update && apt-get install -y git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential libboost-all-dev
+# Install dlib via pip (Python 3.11 should be available now)
+RUN pip install --no-cache-dir dlib
 
+# Set up user
 RUN useradd -m -u 1000 user
-
 USER user
 
+# Set environment variables
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    PATH=/home/user/.local/bin:$PATH \
     PYTHONPATH=$HOME/app \
-	PYTHONUNBUFFERED=1 \
-	GRADIO_ALLOW_FLAGGING=never \
-	GRADIO_NUM_PORTS=1 \
-	GRADIO_SERVER_NAME=0.0.0.0 \
-	GRADIO_THEME=huggingface \
+    PYTHONUNBUFFERED=1 \
+    GRADIO_ALLOW_FLAGGING=never \
+    GRADIO_NUM_PORTS=1 \
+    GRADIO_SERVER_NAME=0.0.0.0 \
+    GRADIO_THEME=huggingface \
     GRADIO_SHARE=False \
-	SYSTEM=spaces
+    SYSTEM=spaces
 
-# Set the working directory to the user's home directory
+# Set working directory
 WORKDIR $HOME/app
 
-RUN git clone -b dev https://github.com/fffiloni/dreamtalk $HOME/app 
-    
+# Clone the app repo
+RUN git clone -b dev https://github.com/fffiloni/dreamtalk $HOME/app
+
+# Download model checkpoints
 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
-    
-# Install dependencies
-RUN pip install --no-cache-dir cmake dlib
-RUN pip install --no-cache-dir urllib3 transformers yacs scipy scikit-image scikit-learn PyYAML Pillow "numpy<2" opencv-python imageio ffmpeg-python av gradio "moviepy<2.0"
 
+# Install other dependencies
+RUN pip install --no-cache-dir \
+    urllib3 transformers yacs scipy scikit-image scikit-learn \
+    PyYAML Pillow numpy opencv-python imageio ffmpeg-python av gradio "moviepy<2"
+
+# Copy app script
 COPY app.py .
 
-# Set the environment variable to specify the GPU device
+# Set CUDA environment variables
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
-# Run your app.py script
-CMD ["python", "app.py"]
+# Run the app
+CMD ["python", "app.py"]