Update Dockerfile

Dockerfile

@@ -1,5 +1,7 @@
 # Builder stage
-FROM nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04 as builder
+FROM nvidia/cuda:12.1.0-cudnn8-runtime-ubuntu22.04
+
+# Add these lines after the FROM statement
 
 # System-level configuration
 ENV DEBIAN_FRONTEND=noninteractive \
@@ -14,18 +16,11 @@ ENV DEBIAN_FRONTEND=noninteractive \
 
 # Install build dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
-    build-essential \
-    python3-dev \
     python3-pip \
-    git \
-    ffmpeg \
-    libsndfile1 \
-    curl \
-    ninja-build \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/* \
-    && rm -rf /var/cache/apt/archives/* \
-    && rm -rf /root/.cache/*
+    && python3 -m pip install --no-cache-dir --upgrade pip
+
 
 # Set up Python
 RUN ln -sf /usr/bin/python3 /usr/bin/python && \
@@ -33,12 +28,8 @@ RUN ln -sf /usr/bin/python3 /usr/bin/python && \
     python3 -m pip install --no-cache-dir pip==23.3.1 setuptools==69.0.3 wheel==0.42.0 && \
     rm -rf /root/.cache/pip/*
 
-# Install PyTorch with CUDA support
-RUN pip3 install --no-cache-dir \
-    torch==2.1.2 \
-    torchaudio==2.1.2 \
-    torchvision==0.16.2 \
-    --extra-index-url https://download.pytorch.org/whl/cu121 && \
+# Replace the pip install command (around line 80) with:
+RUN python3 -m pip install --no-cache-dir -r requirements.txt && \
     rm -rf /root/.cache/pip/* && \
     find /usr/local/lib/python3.* -name '*.pyc' -delete && \
     find /usr/local/lib/python3.* -name '__pycache__' -exec rm -r {} +