Update Dockerfile

Dockerfile

@@ -17,10 +17,14 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 # Copy requirements FIRST for better caching
 COPY requirements.txt .
 
-# Install PyTorch CPU first, then other dependencies
+# Install CPU PyTorch FIRST using the correct method
 RUN pip install --no-cache-dir \
-    torch==2.8.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cpu \
-    && pip install --no-cache-dir -r requirements.txt
+    torch==2.8.0+cpu \
+    torchaudio==2.8.0+cpu \
+    --index-url https://download.pytorch.org/whl/cpu
+
+# Now install the rest of the dependencies
+RUN pip install --no-cache-dir -r requirements.txt
 
 # Clone neutts-air
 RUN git clone --depth 1 https://github.com/neuphonic/neutts-air.git