Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,7 @@
 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
@@ -8,6 +9,8 @@ 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