Update README.md

README.md

@@ -136,3 +136,18 @@ pii_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_str
 
 text = "John Doe's email is john.doe@example.com and his phone number is +1-202-555-0173."
 print(pii_pipeline(text))
+```
+
+
+## Citation
+
+If you use this model, please cite the original DistilBERT paper:
+
+**BibTeX:**
+```bibtex
+@article{sanh2019distilbert,
+  title={DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter},
+  author={Sanh, Victor and Debut, Lysandre and Chaumond, Julien and Wolf, Thomas},
+  journal={arXiv preprint arXiv:1910.01108},
+  year={2019}
+}