lbiester commited on
Commit
e2adb60
·
verified ·
1 Parent(s): 08c539e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def predict_sentiment(text, model):
13
  if model == "finiteautomata/bertweet-base-sentiment-analysis":
14
  pipe = pipeline("text-classification", model="finiteautomata/bertweet-base-sentiment-analysis")
15
  out = pipe(text, return_all_scores=True)
16
- return {pred["label"]: pred["score"] for pred in out}
17
  elif model == "vader":
18
  nltk.download('vader_lexicon')
19
  sia = SentimentIntensityAnalyzer()
 
13
  if model == "finiteautomata/bertweet-base-sentiment-analysis":
14
  pipe = pipeline("text-classification", model="finiteautomata/bertweet-base-sentiment-analysis")
15
  out = pipe(text, return_all_scores=True)
16
+ return {pred["label"]: pred["score"] for pred in out[0]}
17
  elif model == "vader":
18
  nltk.download('vader_lexicon')
19
  sia = SentimentIntensityAnalyzer()