Update Dockerfile

Dockerfile

@@ -30,20 +30,23 @@ ENV HF_HOME=/models/huggingface \
     HUGGINGFACE_HUB_CACHE=/models/huggingface \
     HF_HUB_CACHE=/models/huggingface
 
-# Create cache dir and set permissions
+# Created cache dir and set permissions
 RUN mkdir -p /models/huggingface && chmod -R 777 /models/huggingface
 
-# --- Preload only N-ATLaS model ---
-# HF_TOKEN is passed securely at runtime (not baked into image)
-ARG HF_TOKEN
-ENV HF_TOKEN=$HF_TOKEN
-
-RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/N-ATLaS', use_auth_token='$HF_TOKEN')" \
+RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='facebook/mms-tts-hau')" \
+ && python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='facebook/mms-tts-eng')" \
+ && python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='facebook/mms-tts-yor')" \
  && find /models/huggingface -name '*.lock' -delete
 
+RUN python -c "from transformers import pipeline; pipeline('text-to-speech', model='facebook/mms-tts-hau')" \
+ && python -c "from transformers import pipeline; pipeline('text-to-speech', model='facebook/mms-tts-eng')" \
+ && python -c "from transformers import pipeline; pipeline('text-to-speech', model='facebook/mms-tts-yor')"
+
 # Copy project files
 COPY . .
 
 EXPOSE 7860
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "1"]
+
+I want Atlas to be loaded first and the rest to load when needed