From 6696e752096178130b47549cd5a72532ab44273a Mon Sep 17 00:00:00 2001
From: "jiahao.he@vtradex.com" <794629435@qq.com>
Date: Tue, 18 Mar 2025 00:16:27 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9C=AC=E5=9C=B0=E5=90=91=E9=87=8F=E5=8C=96?=
 =?UTF-8?q?=20docker=E9=95=9C=E5=83=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 script/docker/localModels/Dockerfile       | 21 +++++++++++++--------
 script/docker/localModels/remade.md        |  4 ++++
 script/docker/localModels/requirements.txt |  8 +++++---
 3 files changed, 22 insertions(+), 11 deletions(-)
 create mode 100644 script/docker/localModels/remade.md

diff --git a/script/docker/localModels/Dockerfile b/script/docker/localModels/Dockerfile
index c988920e..0f1e6063 100644
--- a/script/docker/localModels/Dockerfile
+++ b/script/docker/localModels/Dockerfile
@@ -1,21 +1,26 @@
-# 使用官方 Python 作为基础镜像
-FROM python:3.8-slim
+# 使用 Python 3.8 slim 作为基础镜像（轻量稳定）
+FROM python:3.10
 
-# 设置工作目录为 /app
+# 设置工作目录
 WORKDIR /app
 
-# 复制当前目录下的所有文件到 Docker 容器的 /app 目录
+# 复制所有文件到容器
 COPY . /app
 
-# 安装应用依赖
-RUN pip install --no-cache-dir -r requirements.txt
+# 更新 pip 并安装 wheel（避免依赖问题）
+RUN pip install --upgrade pip wheel
+ 
+# 使用阿里云 PyPI 镜像安装依赖，并添加 trusted-host
+RUN pip install -i https://mirrors.aliyun.com/pypi/simple \
+    --trusted-host mirrors.aliyun.com \
+    --no-cache-dir -r requirements.txt
 
-# 暴露 Flask 应用使用的端口
+# 暴露 Flask 端口
 EXPOSE 5000
 
 # 设置环境变量
 ENV FLASK_APP=app.py
 ENV FLASK_RUN_HOST=0.0.0.0
 
-# 启动 Flask 应用
+# 运行 Flask
 CMD ["flask", "run", "--host=0.0.0.0"]
diff --git a/script/docker/localModels/remade.md b/script/docker/localModels/remade.md
new file mode 100644
index 00000000..d6af3d2b
--- /dev/null
+++ b/script/docker/localModels/remade.md
@@ -0,0 +1,4 @@
+1.下载镜像
+docker pull registry.cn-hangzhou.aliyuncs.com/hejh-docker/localmodel:0.1.1
+2. 启动 
+docker run -p 5000:5000  <imageId>
\ No newline at end of file
diff --git a/script/docker/localModels/requirements.txt b/script/docker/localModels/requirements.txt
index c1e1b50b..2d62483a 100644
--- a/script/docker/localModels/requirements.txt
+++ b/script/docker/localModels/requirements.txt
@@ -1,3 +1,5 @@
-Flask==2.0.3
-sentence-transformers==2.2.0
-scikit-learn==0.24.2
+Flask==3.1.0
+sentence-transformers==3.4.1
+scikit-learn==1.6.1
+werkzeug>=3.1
+