diff --git a/Dockerfile b/Dockerfile
index 8818a68..46ab57f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -9,14 +9,14 @@ ENV PYTHONUNBUFFERED=1 \
     # 设置时区
     TZ=Asia/Shanghai \
     # 设置pip镜像地址，加快安装速度
-    PIP_INDEX_URL=http://mirrors.aliyun.com/pypi/simple
+    PIP_INDEX_URL=https://pypi.tuna.tsinghua.edu.cn/simple
 
 # 将当前目录内容复制到容器的/app内
 COPY . /app
 
 # 安装任何需要的包
 RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo '$TZ' > /etc/timezone \
-    && pip install --no-cache-dir --trusted-host mirrors.aliyun.com -r requirements.txt \
+    && pip install --no-cache-dir -r requirements.txt \
     && pip uninstall -y onnxruntime onnxruntime-gpu \
     && pip install onnxruntime-gpu==1.18.0 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
 
diff --git a/requirements.txt b/requirements.txt
index f408f85..14e0c0f 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,5 @@
+pyyaml==6.0.1
+
 numpy==1.26.4
 onnxconverter-common==1.14.0
 opencv-python==4.6.0.66