Vasanth6 commited on
Commit
20522f3
·
1 Parent(s): c45a4b0

Cache public models at build time in Dockerfile to prevent runtime hangs

Browse files
Files changed (1) hide show
  1. Dockerfile +10 -0
Dockerfile CHANGED
@@ -20,6 +20,16 @@ RUN pip install --no-cache-dir .
20
  RUN mkdir -p /app/nltk_data && \
21
  python -c "import nltk; nltk.download('punkt', download_dir='/app/nltk_data'); nltk.download('punkt_tab', download_dir='/app/nltk_data'); nltk.download('stopwords', download_dir='/app/nltk_data')"
22
 
 
 
 
 
 
 
 
 
 
 
23
  # Copy app code
24
  COPY . .
25
 
 
20
  RUN mkdir -p /app/nltk_data && \
21
  python -c "import nltk; nltk.download('punkt', download_dir='/app/nltk_data'); nltk.download('punkt_tab', download_dir='/app/nltk_data'); nltk.download('stopwords', download_dir='/app/nltk_data')"
22
 
23
+ # Pre-download public HF embedding models at build time (Nomic and Qwen)
24
+ # Note: we exclude the gated google/gemma-3n-embedding-exp to prevent build failures.
25
+ RUN python -c "from huggingface_hub import snapshot_download; \
26
+ snapshot_download(repo_id='nomic-ai/nomic-embed-text-v1.5'); \
27
+ snapshot_download(repo_id='Qwen/Qwen3-Embedding-0.6B')"
28
+
29
+ # Pre-download public HF Qwen LLM model at build time
30
+ RUN python -c "from huggingface_hub import snapshot_download; \
31
+ snapshot_download(repo_id='Qwen/Qwen2.5-1.5B-Instruct')"
32
+
33
  # Copy app code
34
  COPY . .
35