update Dockerfile

This commit is contained in:
Qing
2022-11-01 09:09:43 +08:00
parent 7147ffac7f
commit db75357659
4 changed files with 18 additions and 10 deletions

View File

@@ -7,8 +7,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
RUN pip install --upgrade pip && \
pip install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install lama-cleaner
ARG version
RUN pip install lama-cleaner==$version
EXPOSE 8080
CMD ["bash"]
CMD ["bash"]