AlbertCAC commited on
Commit
2694ac7
·
1 Parent(s): d34fd76
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -12,7 +12,7 @@ RUN pip install --no-cache-dir huggingface_hub
12
 
13
  RUN mkdir -p /app/output && \
14
  python -c "from huggingface_hub import hf_hub_download; \
15
- hf_hub_download(repo_id='Albert-CAC/lite_DETECTIVE', filename='lite.pth', local_dir='/app/output', local_dir_use_symlinks=False)"
16
 
17
 
18
  WORKDIR /app
 
12
 
13
  RUN mkdir -p /app/output && \
14
  python -c "from huggingface_hub import hf_hub_download; \
15
+ hf_hub_download(repo_id='Albert-CAC/lite_DETECTIVE', filename='lited_best.pth', local_dir='/app/output', local_dir_use_symlinks=False)"
16
 
17
 
18
  WORKDIR /app