Update Dockerfile

Dockerfile

@@ -30,28 +30,36 @@ RUN pip install --no-cache-dir \
     torchcodec==0.6.0 \
     --index-url https://download.pytorch.org/whl/cu128
 
-# Copy and install transformers-specific requirements (torch already installed above, pip will skip it)
-COPY transformers/requirements.txt /app/transformers/requirements.txt
-RUN pip install --no-cache-dir -r transformers/requirements.txt
+# Copy and install common requirements (torch already installed above, pip will skip it)
+COPY requirements/requirements.txt /app/requirements/requirements.txt
+RUN pip install --no-cache-dir -r requirements/requirements.txt
 
-# Install additional dependencies for specific model families
-# - Voxtral requires mistral-common[audio]
-# - soundfile for audio I/O
+# Install NeMo toolkit with ASR extras
 RUN pip install --no-cache-dir \
-    "mistral-common[audio]>=1.9.0" \
-    soundfile
+    "nemo_toolkit[asr]==2.7.2"
 
-# Copy only the required files
-COPY transformers/ /app/transformers/
-COPY normalizer/ /app/normalizer/
+# Install SALM dependencies (lhotse for audio loading, transformers for GenerationConfig)
+RUN pip install --no-cache-dir \
+    lhotse \
+    sentencepiece
+
+# CUDA-accelerated TDT and RNN-T decoding
+RUN pip install --no-cache-dir \
+    "cuda-python>=12.4"
+
+# Force soundfile backend for datasets audio decoding (avoids torchcodec/FFmpeg issues)
+ENV HF_AUDIO_DECODER_BACKEND=soundfile
+
+# Copy the full repository
+COPY . /app
 
-# Default working directory for running transformers scripts
-WORKDIR /app/transformers
+# Default working directory for running nemo scripts
+WORKDIR /app/nemo_asr
 
 # Default entrypoint
 ENTRYPOINT ["bash"]
 
 # Keep-alive CMD so the Space runtime stays healthy. HF Jobs and `docker run`
-# override this with their own command (e.g. `run_cohere.sh`).
+# override this with their own command (e.g. `run_parakeet.sh`).
 EXPOSE 7860
 CMD ["-c", "python3 -m http.server 7860"]