miibanl commited on
Commit
d644943
·
verified ·
1 Parent(s): 0e933cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -10,7 +10,12 @@ labels = learner.dls.vocab
10
  # Definimos una función que se encarga de llevar a cabo las predicciones
11
  def predict(text):
12
  pred, pred_idx, probs = learner.predict(text)
13
- return(pred[0]);
 
 
 
 
 
14
 
15
 
16
 
 
10
  # Definimos una función que se encarga de llevar a cabo las predicciones
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