FinsightX / tools /sentiment_tool.py
zykrix
Initial commit of FinSightX
bb4e28f
raw
history blame contribute delete
240 Bytes
from transformers import pipeline
sentiment_pipe = pipeline("sentiment-analysis", model="ProsusAI/finbert")
def analyze_sentiment(text: str) -> str:
result = sentiment_pipe(text)
return result[0]["label"] if result else "NEUTRAL"