ink85 commited on
Commit
9a7ca89
·
verified ·
1 Parent(s): 88548da

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +11 -0
src/streamlit_app.py CHANGED
@@ -1,5 +1,16 @@
1
  import os
2
 
 
 
 
 
 
 
 
 
 
 
 
3
  # Redirect Hugging Face cache to writable directory
4
  os.environ["HF_HOME"] = "/app/.cache/huggingface"
5
  os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/transformers"
 
1
  import os
2
 
3
+ # Use /tmp (always writable in Hugging Face Spaces & Docker)
4
+ os.environ["HF_HOME"] = "/tmp/huggingface"
5
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
6
+ os.environ["HF_HUB_CACHE"] = "/tmp/hfhub"
7
+
8
+ # Create directories if not exist
9
+ os.makedirs(os.environ["HF_HOME"], exist_ok=True)
10
+ os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
11
+ os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
12
+
13
+
14
  # Redirect Hugging Face cache to writable directory
15
  os.environ["HF_HOME"] = "/app/.cache/huggingface"
16
  os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/transformers"