Jaja-09 commited on
Commit
e9de246
·
1 Parent(s): a2a5828

fix(build): set NLTK_DATA to writable path and pre-download punkt

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -2
Dockerfile CHANGED
@@ -15,8 +15,10 @@ ARG HF_MODEL_REPO=Jaja-09/authorcheck-model
15
  # Download model snapshot from HF model repo
16
  RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='${HF_MODEL_REPO}', local_dir='/app/model')"
17
 
18
- # Ensure NLTK tokenizer data is present
19
- RUN python -c "import nltk; nltk.download('punkt', quiet=True)"
 
 
20
 
21
  EXPOSE 7860
22
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
 
15
  # Download model snapshot from HF model repo
16
  RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='${HF_MODEL_REPO}', local_dir='/app/model')"
17
 
18
+ # Pre-download NLTK data to a writable path and set env
19
+ ENV NLTK_DATA=/app/nltk_data
20
+ RUN mkdir -p /app/nltk_data && \
21
+ python -c "import nltk; nltk.download('punkt', download_dir='/app/nltk_data', quiet=True)"
22
 
23
  EXPOSE 7860
24
  CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]