Update Dockerfile

Dockerfile

@@ -18,42 +18,44 @@ RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86
 # Install CUDA Toolkit and cuDNN (with version matching the base image)
 RUN apt-get install -y cuda-toolkit-12-1 libcudnn8 libcudnn8-dev
 
-# Set CUDA paths for build environment
-ENV PATH=/usr/local/cuda/bin:$PATH \
+# Create user first so we can set up their home directory
+RUN useradd -m -u 1000 user
+
+# Set environment variables including HOME
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:/usr/local/cuda/bin:$PATH \
     CUDA_TOOLKIT_ROOT_DIR=/usr/local/cuda \
     CUDA_HOME=/usr/local/cuda \
     LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH \
     CMAKE_PREFIX_PATH="/usr/local/cuda:/usr/lib/x86_64-linux-gnu/" \
     TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0;8.6;8.9;9.0"
 
-# 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 cmake
+# Create app directory structure but don't switch to user yet
+RUN mkdir -p /home/user/app/checkpoints && \
+    chown -R user:user /home/user
 
-# Create app directory structure first
-RUN mkdir -p /app/checkpoints
+# Clone the repository directly into the app directory (as root for now)
+RUN git clone -b dev https://github.com/fffiloni/dreamtalk /home/user/app && \
+    chown -R user:user /home/user/app
 
-# Clone the repository
-RUN git clone -b dev https://github.com/fffiloni/dreamtalk /app
+# Download model checkpoints (as root)
+RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O /home/user/app/checkpoints/denoising_network.pth && \
+    wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O /home/user/app/checkpoints/renderer.pt && \
+    chown -R user:user /home/user/app/checkpoints
 
-# Download model checkpoints
-RUN wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/denoising_network.pth -O /app/checkpoints/denoising_network.pth && \
-    wget https://huggingface.co/camenduru/dreamtalk/resolve/main/damo/dreamtalk/checkpoints/renderer.pt -O /app/checkpoints/renderer.pt
+# Switch to user for pip installations
+USER user
+WORKDIR $HOME
+
+# 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 cmake
 
 # Explicitly set environment variables for dlib compilation
 ENV DLIB_USE_CUDA=1 \
     DLIB_JPEG_SUPPORT=1 \
-    DLIB_PNG_SUPPORT=1
-
-# Install dlib with CUDA support
-RUN pip install --no-cache-dir dlib==19.24.0
-
-# Create a user and set ownership
-RUN useradd -m -u 1000 user && \
-    chown -R user:user /app
-
-# Set environment variables for runtime
-ENV PYTHONPATH=/app \
+    DLIB_PNG_SUPPORT=1 \
+    PYTHONPATH=$HOME/app \
     PYTHONUNBUFFERED=1 \
     GRADIO_ALLOW_FLAGGING=never \
     GRADIO_NUM_PORTS=1 \
@@ -64,12 +66,14 @@ ENV PYTHONPATH=/app \
     CUDA_DEVICE_ORDER=PCI_BUS_ID \
     CUDA_VISIBLE_DEVICES=0
 
-# Switch to user
-USER user
-WORKDIR /app
+# Install dlib with CUDA support
+RUN pip install --no-cache-dir dlib==19.24.0
+
+# Set working directory to app
+WORKDIR $HOME/app
 
 # Copy application file (assuming it's in the build context)
-COPY --chown=user:user app.py /app/
+COPY --chown=user:user app.py $HOME/app/
 
 # Run the application
 CMD ["python", "app.py"]