Update app.py

src/app.py

@@ -4,6 +4,23 @@ import time
 import os
 from pathlib import Path
 
+# -----------------
+# Fix Streamlit Permission Issues
+# -----------------
+# 在 HF Space 中，将 Streamlit 配置目录设置到可写位置
+if os.environ.get('SPACE_ID'):
+    os.environ['STREAMLIT_SERVER_FILE_WATCHER_TYPE'] = 'none'
+    os.environ['STREAMLIT_BROWSER_GATHER_USAGE_STATS'] = 'false'
+    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'
+    
+    # 设置可写的配置目录
+    streamlit_dir = Path('/tmp/.streamlit')
+    streamlit_dir.mkdir(exist_ok=True, parents=True)
 
 # -----------------
 # Page Configuration
@@ -25,7 +42,9 @@ def load_model(from_pretrained, base_model, cache_dir, device):
     """
     is_hf_space = os.environ.get('SPACE_ID') is not None
     if is_hf_space:
-        cache_dir = str(Path.home() / '.cache' / 'huggingface')
+        cache_dir = '/tmp/huggingface_cache'
+        os.makedirs(cache_dir, exist_ok=True)
+
         device = 'cpu'
         print("Using **CPU** now!")