Update src/streamlit_app.py

src/streamlit_app.py

@@ -558,9 +558,12 @@ with st.sidebar:
 def load_financial_models():
     """Load multiple financial sentiment models"""
     try:
+        cache_dir = "/tmp/huggingface"
+        os.makedirs(cache_dir, exist_ok=True)
+        
         # FinBERT for financial sentiment
-        finbert_tokenizer = AutoTokenizer.from_pretrained("yiyanghkust/finbert-tone")
-        finbert_model = AutoModelForSequenceClassification.from_pretrained("yiyanghkust/finbert-tone")
+        finbert_tokenizer = AutoTokenizer.from_pretrained("yiyanghkust/finbert-tone", cache_dir=cache_dir)
+        finbert_model = AutoModelForSequenceClassification.from_pretrained("yiyanghkust/finbert-tone", cache_dir=cache_dir)
         
         # Financial NER for entity extraction
         ner_pipeline = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", aggregation_strategy="simple")