Update src/streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,15 @@ import pandas as pd
 import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
+# Use a writable cache directory on Spaces
+os.environ["HF_HOME"] = "/data"
+os.environ["TRANSFORMERS_CACHE"] = "/data/transformers"
+os.environ["HF_HUB_CACHE"] = "/data/hub"
+
+os.makedirs("/data/transformers", exist_ok=True)
+os.makedirs("/data/hub", exist_ok=True)
+
+
 HF_MODEL_ID = "ma7583/pvs_oracle"
 
 # ---------------------------------------------------------------------