Update README.md

README.md

@@ -16,7 +16,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert-base-parsbert-uncased-ncbi_disease
 
-This model is a fine-tuned version of [HooshvareLab/bert-base-parsbert-uncased](https://huggingface.co/HooshvareLab/bert-base-parsbert-uncased) on an unknown dataset.
+This model is a fine-tuned version of [HooshvareLab/bert-base-parsbert-uncased](https://huggingface.co/HooshvareLab/bert-base-parsbert-uncased) on the [ncbi-persian](https://huggingface.co/datasets/Amir13/ncbi-persian) dataset.
 It achieves the following results on the evaluation set:
 - Loss: 0.1018
 - Precision: 0.8192
@@ -76,3 +76,19 @@ The following hyperparameters were used during training:
 - Pytorch 1.13.1+cu116
 - Datasets 2.9.0
 - Tokenizers 0.13.2
+
+### Citation
+If you used the datasets and models in this repository, please cite it.
+
+```bibtex
+@misc{https://doi.org/10.48550/arxiv.2302.09611,
+  doi = {10.48550/ARXIV.2302.09611},
+  url = {https://arxiv.org/abs/2302.09611},
+  author = {Sartipi, Amir and Fatemi, Afsaneh},
+  keywords = {Computation and Language (cs.CL), Artificial Intelligence (cs.AI), FOS: Computer and information sciences, FOS: Computer and information sciences},
+  title = {Exploring the Potential of Machine Translation for Generating Named Entity Datasets: A Case Study between Persian and English},
+  publisher = {arXiv},
+  year = {2023},
+  copyright = {arXiv.org perpetual, non-exclusive license}
+}
+```