Update README.md
Browse files
README.md
CHANGED
@@ -117,7 +117,7 @@ with torch.no_grad():
|
|
117 |
predictions = torch.argmax(outputs.logits, dim=2)
|
118 |
|
119 |
tokens = tokenizer.convert_ids_to_tokens(inputs["input_ids"][0])
|
120 |
-
offset_mapping = offset_mapping[0].tolist()
|
121 |
predicted_labels = [model.config.id2label[pred.item()] for pred in predictions[0]]
|
122 |
word_ids = inputs.word_ids(batch_index=0)
|
123 |
|
|
|
117 |
predictions = torch.argmax(outputs.logits, dim=2)
|
118 |
|
119 |
tokens = tokenizer.convert_ids_to_tokens(inputs["input_ids"][0])
|
120 |
+
offset_mapping = offset_mapping[0].tolist()
|
121 |
predicted_labels = [model.config.id2label[pred.item()] for pred in predictions[0]]
|
122 |
word_ids = inputs.word_ids(batch_index=0)
|
123 |
|