Update Dockerfile

Dockerfile

@@ -1,60 +1,43 @@
-# Use Ubuntu 22.04 as base (Python 3.11 is default)
-FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies
-RUN apt-get update && apt-get install -y \
-    python3.11 python3.11-dev python3.11-venv python3-pip \
-    git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev \
-    build-essential cmake libboost-all-dev \
-    && ln -sf /usr/bin/python3.11 /usr/bin/python \
-    && ln -sf /usr/bin/pip3 /usr/bin/pip
+# Set the MKL_THREADING_LAYER environment variable to GNU
+ENV MKL_THREADING_LAYER=GNU
 
-# Verify Python version
-RUN python --version
+RUN apt-get update && apt-get install -y git wget libgl1-mesa-glx libglib2.0-0 ffmpeg libx264-dev build-essential cmake
 
-# Install a compatible dlib wheel (manylinux_x86_64)
-RUN pip install --no-cache-dir \
-    "dlib==19.10"
-    
-# 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 working directory
+# Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
-# Clone the app repo
-RUN git clone -b dev https://github.com/fffiloni/dreamtalk $HOME/app
-
-# Download model checkpoints
+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
+    
+# Install dependencies
+RUN pip install --no-cache-dir "urllib3 transformers dlib yacs scipy scikit-image scikit-learn PyYAML Pillow numpy<2 opencv-python imageio ffmpeg-python av moviepy<2 gradio"
 
-# 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 CUDA environment variables
+# Set the environment variable to specify the GPU device
 ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
 ENV CUDA_VISIBLE_DEVICES=0
 
-# Run the app
-CMD ["python", "app.py"]
+# Run your app.py script
+CMD ["python", "app.py"]