Update app.py

app.py

@@ -7,6 +7,11 @@ from sentence_transformers import SentenceTransformer
 from huggingface_hub import hf_hub_download
 import faiss
 import time
+import pathlib
+
+# 创建安全缓存目录（在用户目录下）
+CACHE_DIR = "/home/user/cache"
+os.makedirs(CACHE_DIR, exist_ok=True)
 
 # 减少内存占用
 os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "max_split_size_mb:32"
@@ -132,6 +137,9 @@ with gr.Blocks() as demo:
 
 # 启动应用
 if __name__ == "__main__":
+    # 确保缓存目录存在
+    pathlib.Path(CACHE_DIR).mkdir(parents=True, exist_ok=True)
+    
     # 先加载必要资源
     print("启动前预加载资源...")
     load_resources()