fix

Dockerfile

@@ -30,13 +30,13 @@ ENV TZ=Asia/Shanghai \
     RVM_LOCAL_REPO=/opt/data/models/RobustVideoMatting \
     RVM_WEIGHTS_PATH=/opt/data/models/torch/hub/checkpoints/rvm_resnet50.pth \
     RVM_MODEL=resnet50 \
-    AUTO_INIT_GFPGAN=true \
-    AUTO_INIT_DDCOLOR=true \
-    AUTO_INIT_REALESRGAN=true \
-    AUTO_INIT_ANIME_STYLE=true \
+    AUTO_INIT_GFPGAN=false \
+    AUTO_INIT_DDCOLOR=false \
+    AUTO_INIT_REALESRGAN=false \
+    AUTO_INIT_ANIME_STYLE=false \
     AUTO_INIT_CLIP=false \
-    AUTO_INIT_RVM=true \
-    AUTO_INIT_REMBG=true \
+    AUTO_INIT_RVM=false \
+    AUTO_INIT_REMBG=false \
     ENABLE_WARMUP=true \
     REALESRGAN_MODEL=realesr-general-x4v3 \
     CELEBRITY_FIND_THRESHOLD=0.87 \

install.sh

@@ -0,0 +1,2 @@
+# pip install -r requirements.txt -i https://pypi.python.org/simple
+pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple