Spaces:
Runtime error
Runtime error
edit label values
Browse files
app.py
CHANGED
@@ -18,6 +18,13 @@ def roberta(data):
|
|
18 |
label = result[0]['label']
|
19 |
score = result[0]['score']
|
20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
return label, score
|
22 |
|
23 |
def getSent(data, model):
|
|
|
18 |
label = result[0]['label']
|
19 |
score = result[0]['score']
|
20 |
|
21 |
+
if(label == 'LABEL_0'):
|
22 |
+
label = 'Negative'
|
23 |
+
elif(label == 'LABEL_1'):
|
24 |
+
label = 'Neutral'
|
25 |
+
else:
|
26 |
+
label = 'Positive'
|
27 |
+
|
28 |
return label, score
|
29 |
|
30 |
def getSent(data, model):
|