AlbertCAC commited on
Commit
3cdea46
·
1 Parent(s): b396b51
Files changed (1) hide show
  1. Dockerfile +5 -0
Dockerfile CHANGED
@@ -7,6 +7,11 @@ RUN useradd -m -u 1000 user
7
  USER user
8
  ENV PATH="/home/user/.local/bin:$PATH"
9
 
 
 
 
 
 
10
  WORKDIR /app
11
 
12
  COPY --chown=user ./requirements.txt requirements.txt
 
7
  USER user
8
  ENV PATH="/home/user/.local/bin:$PATH"
9
 
10
+ RUN mkdir -p /app/output && \
11
+ python -c "from huggingface_hub import hf_hub_download; \
12
+ hf_hub_download(repo_id='Albert-CAC/lite.pth', filename='lite.pth', local_dir='/app/output', local_dir_use_symlinks=False)"
13
+
14
+
15
  WORKDIR /app
16
 
17
  COPY --chown=user ./requirements.txt requirements.txt