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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,9 +10,9 @@ 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
- if(pred[0]==0):
14
  return "negative"
15
- elif(pred[0]==1):
16
  return "neutral"
17
  else:
18
  return "positive"
 
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"