Update Dockerfile

Dockerfile

@@ -10,6 +10,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3-pip \
     python3-dev \
     git \
+    gcc \
+    g++ \
     libsndfile1 \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
@@ -40,18 +42,11 @@ RUN pip install --no-cache-dir \
     num2words \
     peft
 
-# Install NeMo toolkit with ASR extras
+# Install Kyutai dependencies (moshi for STT model, julius for audio resampling)
 RUN pip install --no-cache-dir \
-    "nemo_toolkit[asr]==2.7.2"
-
-# 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"
+    "moshi>=0.2.6" \
+    julius \
+    soundfile
 
 # Force soundfile backend for datasets audio decoding (avoids torchcodec/FFmpeg issues)
 ENV HF_AUDIO_DECODER_BACKEND=soundfile
@@ -63,6 +58,6 @@ COPY . /app
 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_parakeet.sh`).
+# override this with their own command (e.g. `run_kyutai.sh`).
 EXPOSE 7860
 CMD ["-c", "python3 -m http.server 7860"]