Spaces:

ancebuc
/

tweet-evaluation-space

Sleeping

ancebuc commited on Apr 4

Commit

2ef5d7f

•

1 Parent(s): 67507f6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from transformers import pipeline
 def greet(tweet):
     classifier = pipeline('text-classification', model='ancebuc/tweet-eval-sentiment')
-    label = classifier(tweet)['label']
     labels = {'LABEL_0': 'Negativo :)', 'LABEL_1': 'Neutro :/', 'LABEL_2': 'Positivo :)'}
     return labels[label]

 def greet(tweet):
     classifier = pipeline('text-classification', model='ancebuc/tweet-eval-sentiment')
+    label = classifier(tweet)[0]['label']
     labels = {'LABEL_0': 'Negativo :)', 'LABEL_1': 'Neutro :/', 'LABEL_2': 'Positivo :)'}
     return labels[label]