dahongj commited on
Commit
2d41505
1 Parent(s): b56d254
Files changed (1) hide show
  1. app.py +1 -7
app.py CHANGED
@@ -33,15 +33,9 @@ def siebert(data):
33
  label = result[0]['label']
34
  score = result[0]['score']
35
 
36
- if(label == 'LABEL_0'):
37
- label = 'Negative'
38
- elif(label == 'LABEL_1'):
39
- label = 'Neutral'
40
- else:
41
- label = 'Positive'
42
-
43
  return label, score
44
 
 
45
  def getSent(data, model):
46
  if(model == 'Bertweet'):
47
  label,score = bertweet(data)
 
33
  label = result[0]['label']
34
  score = result[0]['score']
35
 
 
 
 
 
 
 
 
36
  return label, score
37
 
38
+
39
  def getSent(data, model):
40
  if(model == 'Bertweet'):
41
  label,score = bertweet(data)