Spaces:

dahongj
/

sentiment-analysis

Runtime error

dahongj commited on Apr 9, 2023

Commit

4b44a9b

•

1 Parent(s): 123b344

edit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,8 @@ import torch
 def bertweet(data):
     specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
     result = specific_model(data)
-    label = result['label']
-    score = result['score']
     return label, score

 def bertweet(data):
     specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
     result = specific_model(data)
+    label = result[0]['label']
+    score = result[0]['score']
     return label, score