engine.py

from transformers import pipeline

class SentimentAnalyzer:
    def __init__(self) -> None: 
        self.analyzer = pipeline("sentiment-analysis", model = "distilbert-base-uncased-finetuned-sst-2-english")
    
    def score_sentiment(self, sentence: str) -> float:
        return self.analyzer(sentence)[0]
    
    def get_sentiment(self, sentence: str) -> str: 
        sentiment_score = self.score_sentiment(sentence)
        return sentiment_score['label']
    
if __name__ == "__main__":
    sentence = "This place is good"
    sentiment_analyzer = SentimentAnalyzer()
    print(sentiment_analyzer.get_sentiment(sentence))