Rub茅n Escobedo commited on
Commit
74e672e
1 Parent(s): d4287a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,8 +9,7 @@ labels = ["sadness", "happiness", "love", "anger", "fear", "surprise"]
9
 
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 text
14
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
15
 
16
  # Creamos la interfaz y la lanzamos.
 
9
 
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(str(text))
 
13
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
14
 
15
  # Creamos la interfaz y la lanzamos.