Update Dockerfile

Dockerfile

@@ -24,6 +24,12 @@ RUN pip install --no-cache-dir \
 RUN pip install git+https://github.com/m-bain/whisperx.git@v3.1.0 \
      --force-reinstall
 RUN pip install --no-cache-dir "numpy<2" 
+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
 
 COPY requirements.txt .
 RUN pip install -r requirements.txt