Update Dockerfile

Dockerfile

@@ -8,6 +8,28 @@
 
 FROM python:3.11-slim
 
+RUN python - <<'PY'
+from huggingface_hub import hf_hub_download
+import os
+
+target = "/home/user/.yarngpt/models"
+os.makedirs(target, exist_ok=True)
+
+hf_hub_download(
+    repo_id="novateur/WavTokenizer-large-speech-75token",
+    filename="wavtokenizer_large_speech_320_24k.ckpt",
+    local_dir=target,
+)
+
+hf_hub_download(
+    repo_id="novateur/WavTokenizer-medium-speech-75token",
+    filename="wavtokenizer_mediumdata_frame75_3s_nq1_code4096_dim512_kmeans200_attn.yaml",
+    local_dir=target,
+)
+
+print("WavTokenizer files downloaded.")
+PY
+
 # Whisper needs ffmpeg to read audio files, this installs it at the
 # system level since it can't come from pip.
 RUN apt-get update && apt-get install -y --no-install-recommends \