Update README.md

README.md

@@ -16,8 +16,8 @@ Sample usage:
 import torch
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-tokenizer = AutoTokenizer.from_pretrained('dicta-il/mT5-XL-HeQ-ft')
-model = AutoModelForSeq2SeqLM.from_pretrained('dicta-il/mT5-XL-HeQ-ft')
+tokenizer = AutoTokenizer.from_pretrained('dicta-il/mt5-xl-heq-ft')
+model = AutoModelForSeq2SeqLM.from_pretrained('dicta-il/mt5-xl-heq-ft')
 
 model.eval()
 
@@ -47,12 +47,19 @@ Output:
 
 ## Citation
 
-If you use `mT5-XL-HeQ-ft` in your research, please cite ```citation to be added```
+If you use `mt5-xl-heq-ft` in your research, please cite ```DictaBERT: A State-of-the-Art BERT Suite for Modern Hebrew```
 
 **BibTeX:**
 
 ```bibtex
-citation to be added
+@misc{shmidman2023dictabert,
+      title={DictaBERT: A State-of-the-Art BERT Suite for Modern Hebrew}, 
+      author={Shaltiel Shmidman and Avi Shmidman and Moshe Koppel},
+      year={2023},
+      eprint={2308.16687},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
 ```
 
 ## License
@@ -68,6 +75,3 @@ This work is licensed under a
 [cc-by-image]: https://i.creativecommons.org/l/by/4.0/88x31.png
 [cc-by-shield]: https://img.shields.io/badge/License-CC%20BY%204.0-lightgrey.svg
 
-
-
-