Spaces:
Running
Running
Update app.py
Browse files
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
|
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
|