Aeon-Avinash commited on
Commit
6cd89c1
·
verified ·
1 Parent(s): 829c00c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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="Falconsai/intent_classification")
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']]