model/sentiment.py

from model.model import Model

class SentimentAnalysis(Model):
    def __init__(self) -> None:
        self.model_name = None
        self.tasks = "sentiment"
        self.load_model(model_name="crypter70/IndoBERT-Sentiment-Analysis" , tasks=self.tasks)
        
    def predict(self, sentences):
        outputs = super().predict(sentences, self.tasks)
        
        return {
            "result": outputs["label"], 
            "score": outputs["score"]
        }
    
sentiment = SentimentAnalysis()