Update README.md

README.md

@@ -18,9 +18,9 @@ Clinical-Longformer consistently out-perform ClinicalBERT across 10 baseline dat
 
 Load the model directly from Transformers:
 ```
-from transformers import AutoTokenizer, AutoModel
-tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-Longformer")
-model = AutoModel.from_pretrained("yikuan8/Clinical-Longformer")
+from transformers import AutoTokenizer, AutoModelForMaskedLM
+tokenizer = AutoTokenizer.from_pretrained("yikuan8/Clinical-Longformer", use_auth_token=True)
+model = AutoModelForMaskedLM.from_pretrained("yikuan8/Clinical-Longformer", use_auth_token=True)
 ```
 
 If you find our implementation helps, please consider citing this :)