Update app.py

app.py

@@ -7,6 +7,13 @@ import os
 from typing import List, Optional
 from datetime import datetime  # 需在文件开头导入
 from fastapi.responses import MarkdownResponse
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
+class CrossEncoderWrapper:
+    def __init__(self):
+        self.model_name = "cross-encoder/ms-marco-MiniLM-L-6-v2"
+        self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
+        self.model = AutoModelForSequenceClassification.from_pretrained(self.model_name)
 
 # 1. 初始化FastAPI应用
 app = FastAPI(