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