Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,22 @@
+import os
+
+# Fix: Set Hugging Face cache to a writable directory
+os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/huggingface"
+os.environ["HF_METRICS_CACHE"] = "/tmp/huggingface"
+
+# Optional but safe
+os.makedirs("/tmp/huggingface", exist_ok=True)
+
+
+
+
+
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
-import os
+
 
 # Load your model from Hugging Face Hub
 model_name = "laiBatool/laiba-spam-classifier-bert"  # replace with your actual model repo name
@@ -10,7 +25,7 @@ model_name = "laiBatool/laiba-spam-classifier-bert"  # replace with your actual
 @st.cache_resource
 def load_model():
      
-    os.environ['HF_HOME'] = './hf_cache'  # Safe cache location in HF Space
+ 
     tokenizer = AutoTokenizer.from_pretrained(model_name)
     model = AutoModelForSequenceClassification.from_pretrained(model_name)
     return tokenizer, model