Update Dockerfile

Dockerfile

@@ -3,16 +3,15 @@ FROM python:3.10-slim
 RUN apt-get update && apt-get install -y ffmpeg libsndfile1 git && rm -rf /var/lib/apt/lists/*
 
 RUN pip install --upgrade pip
-
-RUN pip install \
-    torch==2.0.1 \
-    torchvision==0.15.2 \
-    torchaudio==2.0.2 \
-    --index-url https://download.pytorch.org/whl/cpu
-
 RUN pip install pyannote.audio==2.1.1
 
 RUN pip install git+https://github.com/m-bain/whisperx.git@v3.1.1
+RUN pip install --no-cache-dir \
+    torch==2.1.2+cpu \
+    torchvision==0.16.2+cpu \
+    torchaudio==2.1.2+cpu \
+    --extra-index-url https://download.pytorch.org/whl/cpu \
+    --force-reinstall
 RUN pip install --no-cache-dir "numpy<2" 
 
 COPY requirements.txt .