Update README.md

README.md

@@ -14,10 +14,10 @@ PS : the tokenizer is the same as the one of the model bert-base-uncased.
 To load the model \& tokenizer :
 
 ````python
-from transformers import BertForMaskedLM, BertTokenizer
+from transformers import AutoModelForMaskedLM, BertTokenizer
 
 model_name = "eli4s/Bert-L12-h384-A6-pruned"
-model = BertForMaskedLM.from_pretrained(model_name)
+model = AutoModelForMaskedLM.from_pretrained(model_name)
 tokenizer = BertTokenizer.from_pretrained(model_name)
 ````