Update app.py

app.py

@@ -6,12 +6,19 @@ import torch
 import os
 from dotenv import load_dotenv
 
-# 设置 Hugging Face 缓存目录
+# 设置多个 Hugging Face 相关环境变量以覆盖默认缓存路径，避免权限问题
 os.environ["HF_HOME"] = "/tmp/.cache/huggingface"
-os.makedirs(os.environ["HF_HOME"], exist_ok=True)
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache/huggingface"
+os.environ["HF_HUB_CACHE"] = "/tmp/.cache/huggingface"
+os.environ["HOME"] = "/tmp"  # 重定向 home 目录到可写路径
 
-# 设置 HOME 环境变量以避免权限问题
-os.environ["HOME"] = "/tmp"
+# 手动创建缓存目录并尝试设置权限
+cache_dir = os.environ["HF_HOME"]
+os.makedirs(cache_dir, exist_ok=True)
+try:
+    os.chmod(cache_dir, 0o777)  # 尝试设置可写权限，如果容器允许
+except PermissionError:
+    pass  # 如果权限不足，忽略并继续（库会使用现有目录）
 
 load_dotenv()  # 加载 .env 文件
 
@@ -26,8 +33,8 @@ if not HF_TOKEN:
 if not API_KEYS:
     raise ValueError("请设置环境变量 API_KEYS")
 
-# 登录 Hugging Face
-login(token=HF_TOKEN)
+# 登录 Hugging Face，添加参数以减少不必要的写入操作
+login(token=HF_TOKEN, add_to_git_credential=False)
 
 # 使用 DeepSeek 7B（开放模型）
 model_name = "deepseek-ai/deepseek-llm-7b"