Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import matplotlib.pyplot as plt
|
|
6 |
|
7 |
|
8 |
def sentiment_analysis(text):
|
9 |
-
analyzer = pipeline("text-classification", model="
|
10 |
# analyzer = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
|
11 |
sentiment = analyzer(text)
|
12 |
return [sentiment[0]['label'], sentiment[0]['score']]
|
|
|
6 |
|
7 |
|
8 |
def sentiment_analysis(text):
|
9 |
+
analyzer = pipeline("text-classification", model="mrm8488/distilroberta-finetuned-financial-news-sentiment-analysis")
|
10 |
# analyzer = pipeline("text-classification", model="distilbert/distilbert-base-uncased-finetuned-sst-2-english")
|
11 |
sentiment = analyzer(text)
|
12 |
return [sentiment[0]['label'], sentiment[0]['score']]
|