rasaenluis3 commited on
Commit
2b6fc14
·
verified ·
1 Parent(s): f7df436

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -24,9 +24,12 @@ def catToValue(cat):
24
  return 'fear'
25
  elif cat == '5':
26
  return 'surprise'
 
 
27
 
28
  # Definimos una función que se encarga de llevar a cabo las predicciones
29
  def predict(texto):
 
30
  pred,pred_idx,probs = learner.predict(texto)
31
  si = {catToValue(labels[i]): float(probs[i]) for i in range(len(labels))}
32
  print(si)
 
24
  return 'fear'
25
  elif cat == '5':
26
  return 'surprise'
27
+ else:
28
+ return 'cagaste'
29
 
30
  # Definimos una función que se encarga de llevar a cabo las predicciones
31
  def predict(texto):
32
+ print(texto)
33
  pred,pred_idx,probs = learner.predict(texto)
34
  si = {catToValue(labels[i]): float(probs[i]) for i in range(len(labels))}
35
  print(si)