animeganv2/Dockerfile

43 lines
1.3 KiB
Docker

# please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges.
FROM docker.io/library/python:3.8.9
RUN apt-get install -y unzip
RUN sed -i 's@//.*archive.ubuntu.com@//mirrors.ustc.edu.cn@g' /etc/apt/sources.list
RUN sed -i 's/security.ubuntu.com/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
RUN --mount=target=/root/packages.txt,source=packages.txt && \
apt-get update && xargs -r -a /root/packages.txt apt-get install -y && \
rm -rf /var/lib/apt/lists/*
WORKDIR /home/user/app
RUN useradd -m -u 1000 user
RUN chown -R 1000.1000 /home/user
RUN mkdir /home/user/app/hub/
RUN mkdir /home/user/app/hub/checkpoints
RUN pip config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/
RUN pip install --no-cache-dir gradio==3.0.9
RUN pip install --no-cache-dir pip==22.3.1
RUN --mount=target=requirements.txt,source=requirements.txt pip install --no-cache-dir -r requirements.txt
ADD gongyoo.jpeg /home/user/app
ADD groot.jpeg /home/user/app
ADD main.zip /home/user/app/hub
ADD hub/checkpoints/face_paint_512_v1.pt /home/user/app/hub/checkpoints/
ADD hub/checkpoints/face_paint_512_v2.pt /home/user/app/hub/checkpoints/
COPY --link --chown=1000 ./ /home/user/app
RUN unzip /home/user/app/hub/main.zip
COPY animeganv2.py /home/user/app
CMD ["python3", "animeganv2.py"]