# please visit https://github.com/xfyun/aiges/releases to get stable and suitable iamges.

FROM public.ecr.aws/iflytek-open/aiges-gpu:11.6-1.17-3.9.13-ubuntu1804-v2.0.0-rc6

WORKDIR /app
#  do this if you are on the chinese server.
RUN pip3 config set global.index-url https://pypi.mirrors.ustc.edu.cn/simple/

RUN mkdir -p requirements

ADD requirements.txt /requirements

RUN pip install -r /requirements/requirements.txt

RUN mkdir -p /app/model && \
    wget https://github.com/AK391/animegan2-pytorch/zipball/main /app/model/

RUN mkdir -p /app/model/checkpoints && \
    wget https://github.com/bryandlee/animegan2-pytorch/raw/main/weights/face_paint_512_v2.pt /app/model/checkpoints/face_paint_512_v2.pt

RUN wget https://github.com/bryandlee/animegan2-pytorch/raw/main/weights/face_paint_512_v1.pt /app/model/checkpoints/face_paint_512_v1.pt