wira.indra commited on
Commit
d453dbe
1 Parent(s): 924ded6

add twitter feature

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -42,8 +42,8 @@ def sentiment_ner(text):
42
  def sentiment_df(df):
43
  text_list = list(df["Text"].astype(str).values)
44
  result = [sentiment_analysis(text) for text in text_list]
45
- df['Label'] = [pred['label'] for pred in result]
46
- df['Score'] = [round(pred['Score'], 3) for pred in result]
47
  return df
48
 
49
 
 
42
  def sentiment_df(df):
43
  text_list = list(df["Text"].astype(str).values)
44
  result = [sentiment_analysis(text) for text in text_list]
45
+ df['Label'] = [list(pred.keys())[0] for pred in result]
46
+ df['Score'] = [round(list(pred.values())[0], 3) for pred in result]
47
  return df
48
 
49