From d219e8dac8c3b6d21c18fff2b6db01d19f30c17a Mon Sep 17 00:00:00 2001
From: SOULOFCINDER <841135647@qq.com>
Date: Tue, 28 Mar 2023 11:27:24 +0800
Subject: [PATCH] update dockerfile

---
 Dockerfile | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 756e0a7..a1fc0b0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -2,15 +2,20 @@
 
 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/
 
-ADD requirements.txt /home/aiges
+RUN mkdir -p requirements
 
-RUN pip install -r /home/aiges/requirements.txt
+ADD requirements.txt /requirements
 
-ADD wrapper /home/aiges
+RUN pip install -r /requirements/requirements.txt
 
-# companion.xfyun.iflytek:6868 is the default config center. It depends on how your athena serving framework built
-CMD ["sh", "-c", "./AIservice -m=0 -c=aiges.toml -s=svcName -u=http://companion.xfyun.iflytek:6868 -p=AIaaS -g=dx"]
\ No newline at end of file
+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
\ No newline at end of file