Update Dockerfile

Dockerfile

@@ -1,69 +1,62 @@
-# --- Base Image ---
+# Use an official PyTorch image with CUDA support as the base image
 FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
-# --- Non-interactive environment ---
+RUN apt-get update && apt-get install -y build-essential
+
+# Install a specific version of pip (e.g., 20.2.4)
+RUN pip install --upgrade pip
+
+# Install Git and system libraries required for OpenGL without interactive prompts
 ENV DEBIAN_FRONTEND=noninteractive
 
-# --- System Dependencies ---
-RUN apt-get update && apt-get install -y \
-    build-essential \
-    git \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    unzip \
-    wget \
-    && rm -rf /var/lib/apt/lists/*
+# Install Git, OpenGL libraries, and libglib2.0
+RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
 
-# --- Upgrade pip ---
-RUN pip install --upgrade pip
+# Set the pip resolver to "legacy"
+RUN pip config set global.resolver legacy
 
-# --- Create a non-root user ---
+#RUN apt-get update && apt-get install -y ninja-build
+
+# Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
+# Switch to the "user" user
 USER user
-WORKDIR /home/user/app
-
-# --- Environment Variables for caches and Gradio ---
+# Set environment variables
 ENV HOME=/home/user \
-    COQUI_TTS_CACHE=/home/user/.cache/tts \
-    XDG_CACHE_HOME=/home/user/.cache \
-    HF_HOME=/home/user/.cache \
-    TRANSFORMERS_CACHE=/home/user/.cache \
-    TORCH_HOME=/home/user/.cache \
-    OMP_NUM_THREADS=1 \
+    CUDA_HOME=/usr/local/cuda \
+    PATH=/home/user/.local/bin:$PATH \
+    LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
+    LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH} \
+    PYTHONPATH=$HOME/app \
     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 \
-    CUDA_HOME=/usr/local/cuda \
-    PATH=/home/user/.local/bin:$PATH \
-    LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH} \
-    LIBRARY_PATH=/usr/local/cuda/lib64/stubs:${LIBRARY_PATH} \
-    CUDA_DEVICE_ORDER=PCI_BUS_ID \
-    CUDA_VISIBLE_DEVICES=0
+    SYSTEM=spaces
 
-# --- Copy app sources ---
-COPY --chown=user:user . .
+# Set the working directory to the user's home directory
+WORKDIR $HOME/app
 
-# --- Make app directories writable ---
-RUN mkdir -p /home/user/.cache /home/user/.local /home/user/app/bark_voices \
-    && chmod -R 777 /home/user/.cache /home/user/.local /home/user/app/bark_voices
+COPY . .
 
-# --- Hack: redirect /root accesses to /home/user ---
 USER root
-RUN rm -rf /root && ln -s /home/user /root
+RUN chmod -R 777 $HOME/app/bark_voices
 USER user
 
-# --- Manual Bark model download to prevent pickle errors ---
-RUN mkdir -p /home/user/.cache/tts_models/multilingual/multi-dataset \
- && cd /home/user/.cache/tts_models/multilingual/multi-dataset \
- && wget -O bark.zip "https://huggingface.co/coqui/tts_models/multilingual/multi-dataset/bark/resolve/main/bark.zip" \
- && unzip bark.zip && rm bark.zip
-
-# --- Install Python dependencies ---
+# Install dependencies
+# Use the new pip resolver and always take the latest version if not specified
 RUN pip install --use-feature=fast-deps -r requirements.txt gradio
 
-# --- Run the app ---
-CMD ["python", "app.py"]
+
+# Update package lists and install other dependencies as needed
+# Ensure that CUDA components are correctly installed and configured
+# Install any other required packages
+
+# Set the environment variable to specify the GPU device
+ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
+ENV CUDA_VISIBLE_DEVICES=0
+
+# Run your app.py script
+CMD ["python", "app.py"]