push

Dockerfile

@@ -47,14 +47,14 @@ RUN python -c "from transformers import pipeline; pipeline('text-to-speech', mod
  && python -c "from transformers import pipeline; pipeline('text-to-speech', model='facebook/mms-tts-yor')"
 
 # Pre-load N-ATLaS model during build
-RUN python -c "import os; from huggingface_hub import snapshot_download; token = os.getenv('HF_TOKEN'); snapshot_download(repo_id='NCAIR1/N-ATLaS', token=token)" \
+RUN python -c "import os; token = os.getenv('HF_TOKEN'); print(f'HF_TOKEN available: {bool(token)}'); print(f'Token length: {len(token) if token else 0}'); from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/N-ATLaS', token=token)" \
  && python -c "import os; from transformers import AutoTokenizer, AutoModelForCausalLM; import torch; token = os.getenv('HF_TOKEN'); tokenizer = AutoTokenizer.from_pretrained('NCAIR1/N-ATLaS', token=token); model = AutoModelForCausalLM.from_pretrained('NCAIR1/N-ATLaS', torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32, device_map='auto' if torch.cuda.is_available() else None, trust_remote_code=True, low_cpu_mem_usage=True, use_cache=True, token=token); print('N-ATLaS model loaded successfully')"
 
 # Pre-download ASR models (will be lazy-loaded at runtime)
-RUN python -c "import os; from huggingface_hub import snapshot_download; token = os.getenv('HF_TOKEN'); snapshot_download(repo_id='NCAIR1/Hausa-ASR', token=token)" \
- && python -c "import os; from huggingface_hub import snapshot_download; token = os.getenv('HF_TOKEN'); snapshot_download(repo_id='NCAIR1/Yoruba-ASR', token=token)" \
- && python -c "import os; from huggingface_hub import snapshot_download; token = os.getenv('HF_TOKEN'); snapshot_download(repo_id='NCAIR1/Igbo-ASR', token=token)" \
- && python -c "import os; from huggingface_hub import snapshot_download; token = os.getenv('HF_TOKEN'); snapshot_download(repo_id='NCAIR1/NigerianAccentedEnglish', token=token)" \
+RUN python -c "import os; token = os.getenv('HF_TOKEN'); print(f'ASR download - HF_TOKEN available: {bool(token)}'); from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/Hausa-ASR', token=token)" \
+ && python -c "import os; token = os.getenv('HF_TOKEN'); from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/Yoruba-ASR', token=token)" \
+ && python -c "import os; token = os.getenv('HF_TOKEN'); from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/Igbo-ASR', token=token)" \
+ && python -c "import os; token = os.getenv('HF_TOKEN'); from huggingface_hub import snapshot_download; snapshot_download(repo_id='NCAIR1/NigerianAccentedEnglish', token=token)" \
  && find /models/huggingface -name '*.lock' -delete
 
 # Copy project files