Merge pull request #60 from davidefiocco/patch-1

Updated quick-start example with `BertForMaskedLM`

Merge pull request #60 from davidefiocco/patch-1
Updated quick-start example with `BertForMaskedLM`
3951c2c1 · Thomas Wolf · GitHub · 21f01964 · ec2c339b · 3951c2c1
Unverified Commit 3951c2c1 authored Nov 28, 2018 by Thomas Wolf Committed by GitHub Nov 28, 2018
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

README.md README.md +1 -1

No files found.
--- a/README.md
+++ b/README.md
@@ -142,7 +142,7 @@ predictions = model(tokens_tensor, segments_tensors)

 # confirm we were able to predict 'henson'
 predicted_index = torch.argmax(predictions[0, masked_index]).item()
-predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])
+predicted_token = tokenizer.convert_ids_to_tokens([predicted_index])[0]
 assert predicted_token == 'henson'
 ```