miibanl commited on
Commit
8d2b036
·
verified ·
1 Parent(s): d644943

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,11 +11,11 @@ labels = learner.dls.vocab
11
  def predict(text):
12
  pred, pred_idx, probs = learner.predict(text)
13
  if(pred[0]==0):
14
- return negative
15
  elif(pred[0]==1):
16
- return neutral
17
  else:
18
- return positive
19
 
20
 
21
 
 
11
  def predict(text):
12
  pred, pred_idx, probs = learner.predict(text)
13
  if(pred[0]==0):
14
+ return "negative"
15
  elif(pred[0]==1):
16
+ return "neutral"
17
  else:
18
+ return "positive"
19
 
20
 
21