dahongj commited on
Commit
e38acc3
1 Parent(s): 747d201

edit label values

Browse files
Files changed (1) hide show
  1. app.py +7 -0
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):