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

fix(build): disable HF_HUB_ENABLE_HF_TRANSFER to avoid hf_transfer requirement

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -11,10 +11,11 @@ COPY app.py /app/app.py
11
  COPY model_handler.py /app/model_handler.py
12
 
13
  ARG HF_MODEL_REPO=Jaja-09/authorcheck-model
14
- ENV HF_HUB_ENABLE_HF_TRANSFER=1
15
 
16
- RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='${HF_MODEL_REPO}', local_dir='/app/model', local_dir_use_symlinks=False)"
 
17
 
 
18
  RUN python -c "import nltk; nltk.download('punkt', quiet=True)"
19
 
20
  EXPOSE 7860
 
11
  COPY model_handler.py /app/model_handler.py
12
 
13
  ARG HF_MODEL_REPO=Jaja-09/authorcheck-model
 
14
 
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