Update README.md

README.md

@@ -107,3 +107,21 @@ The computational performance of ProkBERT PhaStyle was evaluated using 1,000 ran
 ## Limitations
 
 ProkBERT PhaStyle is specifically designed for **binary classification** of phage lifestyles (virulent vs. temperate) and does not handle non-phage sequences. It is recommended to use this model in conjunction with upstream pipelines that identify phage sequences. For large-scale inference, **GPU support** is strongly advised.
+
+# Citing this work
+
+If you use the data in this package, please cite:
+
+```bibtex
+@Article{ProkBERT2024,
+  author  = {Ligeti, Balázs and Szepesi-Nagy, István and Bodnár, Babett and Ligeti-Nagy, Noémi and Juhász, János},
+  journal = {Frontiers in Microbiology},
+  title   = {{ProkBERT} family: genomic language models for microbiome applications},
+  year    = {2024},
+  volume  = {14},
+  URL={https://www.frontiersin.org/articles/10.3389/fmicb.2023.1331233},       
+	DOI={10.3389/fmicb.2023.1331233},      
+	ISSN={1664-302X}
+}
+```
+