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

class RobertaLarge(BaseModel):
    """
    A class for performing sentiment analysis using the FinBERT model.
    
    https://huggingface.co/Farshid/roberta-large-financial-phrasebank-allagree1
    """
    
    def __init__(self):
        super().__init__("RobertaLarge")
        self.tokenizer = AutoTokenizer.from_pretrained("Farshid/roberta-large-financial-phrasebank-allagree1")
        self.model = AutoModelForSequenceClassification.from_pretrained("Farshid/roberta-large-financial-phrasebank-allagree1")
        self.model.to(self.device)
        self.model.eval()