vrashad commited on
Commit
25cd922
·
verified ·
1 Parent(s): 10be618

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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