Update Dockerfile

Dockerfile

@@ -2,33 +2,32 @@ FROM python:3.10-slim
 
 WORKDIR /app
 
-# Install system dependencies required for building and audio processing
+# Install system dependencies (added ffmpeg just in case torchaudio needs it for your audio types)
 RUN apt-get update && apt-get install -y \
-    git bash wget build-essential libsndfile1 \
+    git bash wget build-essential libsndfile1 ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
-# Force install PyTorch CPU version to save massive amounts of build time and space
-RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
-
-# Clone and install the custom NeMo fork required by IndicConformerASR
+# Combine the NeMo install and the PyTorch CPU override into ONE step.
+# This ensures matching versions (2.1.2) and prevents Docker from caching massive GPU binaries.
 RUN git clone https://github.com/AI4Bharat/NeMo.git \
     && cd NeMo \
     && git checkout nemo-v2 \
     && bash reinstall.sh \
+    && pip uninstall -y torch torchvision torchaudio \
+    && pip install --no-cache-dir torch==2.1.2+cpu torchvision==0.16.2+cpu torchaudio==2.1.2+cpu --index-url https://download.pytorch.org/whl/cpu \
     && cd ..
 
 # Install FastAPI, Piper TTS, and patch the known dependency issues
 RUN pip install --no-cache-dir "numpy<2.0" huggingface_hub==0.23.2 fastapi uvicorn python-multipart "piper-tts==1.2.0"
 
-# Fetch the Piper Nepali Chitwan model directly during the Docker build
+# Fetch the Piper Nepali Chitwan model directly
 RUN wget -q -O chitwan.onnx "https://huggingface.co/rhasspy/piper-voices/resolve/main/ne/ne_NP/chitwan/medium/ne_NP-chitwan-medium.onnx?download=true"
 RUN wget -q -O chitwan.onnx.json "https://huggingface.co/rhasspy/piper-voices/resolve/main/ne/ne_NP/chitwan/medium/ne_NP-chitwan-medium.onnx.json?download=true"
 
 # Copy the FastAPI app into the container
 COPY app.py /app/app.py
 
-# Hugging Face Spaces route traffic through port 7860
 EXPOSE 7860
 
 # Start the asynchronous Uvicorn server
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]