更新 Dockerfile，安装 Spacy 及其模型；在 preprocess.py 中添加模型下载处理

Dockerfile

@@ -10,6 +10,8 @@ COPY . /app
 # 安装所需的 Python 包
 RUN pip install --no-cache-dir --upgrade -r /app/requirements.txt
 
+# 安装必要依赖
+RUN pip install --no-cache-dir spacy
 
 # 公开 Gradio 默认的端口 7860
 EXPOSE 7860
@@ -29,6 +31,10 @@ WORKDIR $HOME/app
 COPY --chown=user . $HOME/app
 
 
+# 下载 Spacy 模型
+RUN python -m spacy download en_core_web_md
+
+
 # Get secret EXAMPLE and output it to /test at buildtime
 RUN --mount=type=secret,id=HF_Token,mode=0444,required=true \
 	cat /run/secrets/HF_Token > $HOME/HF_Token

preprocess.py

@@ -34,7 +34,13 @@ from us_stock import *
 #spacy.require_gpu()
 
 # 加载模型
-nlp = spacy.load("en_core_web_md")
+try:
+    nlp = spacy.load("en_core_web_md")
+except OSError:
+    print("Downloading model 'en_core_web_md'...")
+    from spacy.cli import download
+    download("en_core_web_md")
+    nlp = spacy.load("en_core_web_md")
 
 # 检查是否使用 GPU
 print("Is NPL GPU used Preprocessing.py:", spacy.prefer_gpu())