Update app.py

src/app.py

@@ -1,6 +1,3 @@
-import streamlit as st
-from FineTune.model import ComputeScore
-import time
 import os
 from pathlib import Path
 
@@ -14,14 +11,22 @@ if os.environ.get('SPACE_ID'):
     os.environ['STREAMLIT_SERVER_ENABLE_CORS'] = 'false'
     
     # 设置 HuggingFace 缓存到可写目录
-    os.environ['HF_HOME'] = '/tmp/huggingface_cache'
-    os.environ['TRANSFORMERS_CACHE'] = '/tmp/huggingface_cache'
-    os.environ['HF_DATASETS_CACHE'] = '/tmp/huggingface_cache'
+    CACHE_DIR = '/tmp/huggingface_cache'
+    os.makedirs(CACHE_DIR, exist_ok=True)
+
+    os.environ['HF_HOME'] = CACHE_DIR
+    os.environ['TRANSFORMERS_CACHE'] = CACHE_DIR
+    os.environ['HF_DATASETS_CACHE'] = CACHE_DIR
+    os.environ['HUGGINGFACE_HUB_CACHE'] = CACHE_DIR
     
     # 设置可写的配置目录
     streamlit_dir = Path('/tmp/.streamlit')
     streamlit_dir.mkdir(exist_ok=True, parents=True)
 
+import streamlit as st
+from FineTune.model import ComputeScore
+import time
+
 # -----------------
 # Page Configuration
 # -----------------