fix: 优化打包脚本

Dockerfile

@@ -8,6 +8,11 @@ RUN python -m pip install paddlepaddle==2.5.2 -i https://pypi.tuna.tsinghua.edu.
 
 RUN cd /app && pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
 
+RUN sed -i 's/deb.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
+RUN sed -i 's/security.debian.org/mirrors.ustc.edu.cn/g' /etc/apt/sources.list
+RUN apt update
+RUN apt install -y libgl1-mesa-glx
+
 WORKDIR /app
 ENTRYPOINT ["python"]
 CMD ["main.py", "--production"]

docker-compose.yaml

@@ -0,0 +1,19 @@
+version: "3"
+
+services:
+  matting:
+    image: registry.cn-hangzhou.aliyuncs.com/tuon-pub/matting-human
+    ports:
+      - 20201:20201
+    volumns:
+      - ./uploads:/app/uploads
+      - ./outputs:/app/outputs
+    restart: always
+    deploy:
+      resources:
+        limits:
+          cpus: '1'
+          memory: 2G
+        reservations:
+          cpus: '0.5'
+          memory: 512M

requirements.txt

@@ -11,5 +11,5 @@ flask~=2.0.3
 six
 scipy
 pillow
-werkzeug
+werkzeug==2.0.2
 wtforms