ura23 commited on
Commit
9ccbf15
·
verified ·
1 Parent(s): 67ae085

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -119,7 +119,7 @@ class Predictor:
119
 
120
  labels = list(zip(self.tag_names, preds[0].astype(float)))
121
  general_res = [x[0] for i, x in enumerate(labels) if i in self.general_indexes and x[1] > general_thresh]
122
- character_res = [x[0] for i, x in enumerate(labels) if i in self.character_indexes and x[1] > character_thresh]
123
  results.append((general_res, character_res))
124
 
125
  return results
 
119
 
120
  labels = list(zip(self.tag_names, preds[0].astype(float)))
121
  general_res = [x[0] for i, x in enumerate(labels) if i in self.general_indexes and x[1] > general_thresh]
122
+ character_res = [x[0] for i, x in enumerate(labels) if i in the character_indexes and x[1] > character_thresh]
123
  results.append((general_res, character_res))
124
 
125
  return results