Update src/streamlit_app.py

src/streamlit_app.py

@@ -47,6 +47,7 @@ import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import torch.nn.functional as F
+import os
 
 st.set_page_config(page_title="FinBERT Sentiment", layout="centered")
 st.title("💰 FinBERT: Financial Sentiment Analysis")
@@ -54,14 +55,17 @@ st.markdown("Модель: `yiyanghkust/finbert-tone` — обучена на ф
 
 @st.cache_resource
 def load_model():
-    tokenizer = AutoTokenizer.from_pretrained("yiyanghkust/finbert-tone")
-    model = AutoModelForSequenceClassification.from_pretrained("yiyanghkust/finbert-tone")
+    # Установка кастомного пути к кэшу
+    cache_dir = "/tmp/huggingface"
+    os.makedirs(cache_dir, exist_ok=True)
+
+    tokenizer = AutoTokenizer.from_pretrained("yiyanghkust/finbert-tone", cache_dir=cache_dir)
+    model = AutoModelForSequenceClassification.from_pretrained("yiyanghkust/finbert-tone", cache_dir=cache_dir)
     return tokenizer, model
 
 tokenizer, model = load_model()
 
-# Ввод текста
-text = st.text_area("Введите финансовую новость, заголовок или отчёт:", height=150)
+text = st.text_area("Введите финансовую новость или отчёт:", height=150)
 
 if st.button("Анализировать тональность") and text.strip():
     inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)