Project / app /model.py
Swathi6's picture
Create app/model.py
a29d505 verified
from transformers import pipeline
# Load the model once on import
sentiment_pipeline = pipeline("sentiment-analysis")
def get_sentiment(text):
result = sentiment_pipeline(text)
return result[0]