add twitter feature

app.py

@@ -2,9 +2,7 @@ from transformers import pipeline
 import matplotlib.pyplot as plt
 import twitter_scraper as ts
 
-
 import gradio as gr
-from gradio.mix import Parallel
 
 pretrained_sentiment = "w11wo/indonesian-roberta-base-sentiment-classifier"
 pretrained_ner = "cahya/bert-base-indonesian-NER"
@@ -42,8 +40,14 @@ def sentiment_ner(text):
 def sentiment_df(df):
     text_list = list(df["Text"].astype(str).values)
     result = [sentiment_analysis(text) for text in text_list]
-    df['Label'] = [list(pred.keys())[0] for pred in result]
-    df['Score'] = [round(list(pred.values())[0], 3) for pred in result]
+    labels = []
+    scores = []
+    for pred in result:
+        idx = list(pred.values()).index(max(list(pred.values())))
+        labels.append(list(pred.keys())[idx])
+        scores.append(round(list(pred.values())[idx], 3))
+    df['Label'] = labels
+    df['Score'] = scores
     return df