docker update for torch

Dockerfile

@@ -4,6 +4,9 @@ FROM python:3.10-slim
 # Set working directory
 WORKDIR /app
 
+RUN git config user.email "[email protected]" && \
+    git config user.name "peace2024"
+
 # Install system-level dependencies
 RUN apt-get update && \
     apt-get install -y ffmpeg libsndfile1 wget curl git && \
@@ -15,6 +18,7 @@ COPY requirements.txt .
 # Upgrade pip and install dependencies
 RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir -r requirements.txt
+    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
 
 # Copy the entire app source code
 COPY . .