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()
|