Update README.md

README.md

@@ -11,23 +11,19 @@ Pretrained BERT Medium language model for Arabic
 
 _If you use this model in your work, please cite this paper:_
 
-<!--```
-@inproceedings{
-  title={KUISAIL at SemEval-2020 Task 12: BERT-CNN for Offensive Speech Identification in Social Media},
-  author={Safaya, Ali and Abdullatif, Moutasem and Yuret, Deniz},
-  booktitle={Proceedings of the International Workshop on Semantic Evaluation (SemEval)},
-  year={2020}
-}
-```-->
-
 ```
-@misc{safaya2020kuisail,
-    title={KUISAIL at SemEval-2020 Task 12: BERT-CNN for Offensive Speech Identification in Social Media},
-    author={Ali Safaya and Moutasem Abdullatif and Deniz Yuret},
-    year={2020},
-    eprint={2007.13184},
-    archivePrefix={arXiv},
-    primaryClass={cs.CL}
+@inproceedings{safaya-etal-2020-kuisail,
+    title = "{KUISAIL} at {S}em{E}val-2020 Task 12: {BERT}-{CNN} for Offensive Speech Identification in Social Media",
+    author = "Safaya, Ali  and
+      Abdullatif, Moutasem  and
+      Yuret, Deniz",
+    booktitle = "Proceedings of the Fourteenth Workshop on Semantic Evaluation",
+    month = dec,
+    year = "2020",
+    address = "Barcelona (online)",
+    publisher = "International Committee for Computational Linguistics",
+    url = "https://www.aclweb.org/anthology/2020.semeval-1.271",
+    pages = "2054--2059",
 }
 ```
 
@@ -59,7 +55,7 @@ You can use this model by installing `torch` or `tensorflow` and Huggingface lib
 from transformers import AutoTokenizer, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained("asafaya/bert-medium-arabic")
-model = AutoModel.from_pretrained("asafaya/bert-medium-arabic")
+model = AutoModelForMaskedLM.from_pretrained("asafaya/bert-medium-arabic")
 ```
 
 ## Results