denis-gordeev
commited on
Commit
•
bbcb533
1
Parent(s):
8fbaf41
Update README.md
Browse files
README.md
CHANGED
@@ -50,6 +50,7 @@ def predict(text:str, glue_tokens=False, output_together=True, glue_words=True):
|
|
50 |
|
51 |
output_tokens = []
|
52 |
output_preds = []
|
|
|
53 |
for i, token in enumerate(input_ids[0]):
|
54 |
if token > 3:
|
55 |
class_ids = (logits[0][i] > 0.5).nonzero()
|
|
|
50 |
|
51 |
output_tokens = []
|
52 |
output_preds = []
|
53 |
+
id_to_label = {int(k): v for k, v in model.config.id2label.items()}
|
54 |
for i, token in enumerate(input_ids[0]):
|
55 |
if token > 3:
|
56 |
class_ids = (logits[0][i] > 0.5).nonzero()
|