src/news_scraper/nlp_models/finbert_tone.py

from transformers import AutoTokenizer, AutoModelForSequenceClassification
from news_scraper.nlp_models.basemodel import BaseModel


class FinBertSentimentAnalyzer_Tone(BaseModel):
    """
    A class for performing sentiment analysis using the FinBERT model.
    
    https://huggingface.co/yiyanghkust/finbert-tone
    """
    # Configure logging


    def __init__(self):
        super().__init__("finbert-tone")
        """Initialize the FinBERT sentiment analyzer."""
        self.tokenizer = AutoTokenizer.from_pretrained("ProsusAI/finbert")
        self.model = AutoModelForSequenceClassification.from_pretrained("ProsusAI/finbert")
        self.model.to(self.device)
        self.model.eval()