masanorihirano commited on
Commit
dc63c9c
·
1 Parent(s): f732d7c
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -1,6 +1,4 @@
1
  FROM docker.io/nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04
2
- COPY --link --chown=1000 ./ /home/user/app
3
- RUN useradd -m -u 1000 user
4
  ENV TZ=Asia/Tokyo
5
  RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
6
  RUN sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && \
@@ -26,6 +24,8 @@ RUN eval "$(pyenv init -)" && \
26
  pip install --no-cache-dir datasets "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1" && \
27
  curl -sSL https://install.python-poetry.org | python -
28
  ENV PATH /root/.local/bin:${PATH}
 
 
29
  WORKDIR /home/user/app
30
 
31
  RUN poetry install
@@ -33,4 +33,4 @@ RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
33
  git config --global credential.helper store && \
34
  huggingface-cli login --token $(cat /run/secrets/HF_TOKEN) --add-to-git-credential
35
  RUN poetry run python model_pull.py
36
- CMD ["env", "poetry", "run", "python", "app.py", "--host", "0.0.0.0", "--port", "7860"]
 
1
  FROM docker.io/nvidia/cuda:11.7.1-cudnn8-devel-ubuntu20.04
 
 
2
  ENV TZ=Asia/Tokyo
3
  RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
4
  RUN sed -i 's http://deb.debian.org http://cdn-aws.deb.debian.org g' /etc/apt/sources.list && \
 
24
  pip install --no-cache-dir datasets "huggingface-hub>=0.12.1" "protobuf<4" "click<8.1" && \
25
  curl -sSL https://install.python-poetry.org | python -
26
  ENV PATH /root/.local/bin:${PATH}
27
+ COPY --link --chown=1000 ./ /home/user/app
28
+ RUN useradd -m -u 1000 user
29
  WORKDIR /home/user/app
30
 
31
  RUN poetry install
 
33
  git config --global credential.helper store && \
34
  huggingface-cli login --token $(cat /run/secrets/HF_TOKEN) --add-to-git-credential
35
  RUN poetry run python model_pull.py
36
+ ENTRYPOINT ["poetry", "run", "python", "app.py", "--host", "0.0.0.0", "--port", "7860"]