Update Dockerfile

Dockerfile

@@ -5,11 +5,45 @@ ENV PYTHONUNBUFFERED=1
 
 WORKDIR /app
 
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+        git \
+        wget \
+        curl \
+        libsndfile1 \
+        ffmpeg \
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
+
+
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
+
+RUN pip install --no-cache-dir \
+    "yarngpt @ git+https://github.com/saheedniyi02/yarngpt.git"
+
+RUN wget -q \
+    https://huggingface.co/novateur/WavTokenizer-medium-speech-75token/resolve/main/wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml \
+    -O /app/wavtokenizer_config.yaml
+
+
+RUN python3 -c "\
+from huggingface_hub import hf_hub_download; \
+hf_hub_download( \
+    repo_id='novateur/WavTokenizer-medium-speech-75token', \
+    filename='wavtokenizer_large_speech_320_24k.ckpt', \
+    local_dir='/app' \
+)"
+
+
+ENV MODEL_ID="saheedniyi/YarnGPT2b"
+ENV WAV_TOKENIZER_CONFIG="/app/wavtokenizer_config.yaml"
+ENV WAV_TOKENIZER_CKPT="/app/wavtokenizer_large_speech_320_24k.ckpt"
+ENV HF_HOME="/app/.cache/huggingface"
+
+
 COPY app ./app
 
 EXPOSE 7860
 
-CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1", "--timeout-keep-alive", "300"]