Update README.md

README.md

@@ -15,7 +15,7 @@ from transformers import *
 import torch
 
 tokenizer = AutoTokenizer.from_pretrained("jeniya/BERTOverflow")
-model = AutoModel.from_pretrained("jeniya/BERTOverflow")
+model = AutoModelForTokenClassification.from_pretrained("jeniya/BERTOverflow")
 
 ```
 
@@ -28,6 +28,7 @@ model = AutoModel.from_pretrained("jeniya/BERTOverflow")
     title={Code and Named Entity Recognition in StackOverflow},
     author={Tabassum, Jeniya  and Maddela, Mounica  and Xu, Wei and Ritter, Alan },
     booktitle = {Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics (ACL)},
+    url={https://www.aclweb.org/anthology/2020.acl-main.443/}
     year = {2020},
 }
 ```