Update README.md

README.md

@@ -3,8 +3,7 @@ license: cc-by-nc-sa-4.0
 pipeline_tag: fill-mask
 language: en
 tags:
-- long_documents
-- hierarchical_transformers
+- long-documents
 datasets:
 - wikipedia
 model-index:
@@ -34,9 +33,10 @@ Note that this model is primarily aimed at being fine-tuned on tasks that use th
 You can use this model directly with a pipeline for masked language modeling:
 
 ```python
-from transformers import pipeline
-mlm_model = pipeline('fill-mask', model='kiddothe2b/hierarchical-transformer-I3-mini-1024', trust_remote_code=True)
-mlm_model("Hello I'm a <mask> model.")
+from transformers import AutoTokenizer, AutoModelforForMaskedLM
+tokenizer = AutoTokenizer.from_pretrained("kiddothe2b/hierarchical-transformer-I3-mini-1024", trust_remote_code=True)
+mlm_model = AutoModelforForMaskedLM(model='kiddothe2b/hierarchical-transformer-I3-mini-1024', trust_remote_code=True)
+
 ```
 
 You can also fine-tun it for SequenceClassification, SequentialSentenceClassification, and MultipleChoice down-stream tasks:
@@ -97,7 +97,8 @@ The following hyperparameters were used during training:
 - Tokenizers 0.11.6
 
 
-##Citing
+## Citing
+
 If you use HAT in your research, please cite [An Exploration of Hierarchical Attention Transformers for Efficient Long Document Classification](https://arxiv.org/abs/xxx)
 
 ```