daledem commited on
Commit
32f9a8b
1 Parent(s): 411ead8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -10,10 +10,7 @@ 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]==1:
14
- return "positiva"
15
- else:
16
- return "negativa"
17
 
18
 
19
  # Creamos la interfaz y la lanzamos.
 
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
 
 
 
14
 
15
 
16
  # Creamos la interfaz y la lanzamos.