sentiment.py

from transformers import pipeline

model = pipeline("sentiment-analysis", model="IDEA-CCNL/Erlangshen-Roberta-110M-Sentiment")

def analyze_sentiment(text: str):
    result = model(text)[0]
    return result["label"], result["score"]