luisvarona commited on
Commit
f2c7fbd
·
verified ·
1 Parent(s): eff8d1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def predict(str):
12
  label = dic_label_score['label']
13
  score = dic_label_score['score']
14
  dicc = {'LABEL_0': 'ABBR', 'LABEL_1': 'ENTY', 'LABEL_2': 'DESC', 'LABEL_3': 'HUM', 'LABEL_4': 'LOC', 'LABEL_5': 'NUM'}
15
- label = label.map(dicc)
16
  return label + ', con score ' + str(score)
17
 
18
  # Creamos la interfaz y la lanzamos.
 
12
  label = dic_label_score['label']
13
  score = dic_label_score['score']
14
  dicc = {'LABEL_0': 'ABBR', 'LABEL_1': 'ENTY', 'LABEL_2': 'DESC', 'LABEL_3': 'HUM', 'LABEL_4': 'LOC', 'LABEL_5': 'NUM'}
15
+ label = dicc[label]
16
  return label + ', con score ' + str(score)
17
 
18
  # Creamos la interfaz y la lanzamos.