Update README.md

README.md

@@ -44,10 +44,18 @@ The core functionality of Architext is taking a string of text and generating a
 # Citation and Related Information
 ## BibTeX entry
 To cite this model:
-TBD
 
-To cite the codebase that trained this model:
+```
+@article{galanos2023architext,
+  title={Architext: Language-Driven Generative Architecture Design},
+  author={Galanos, Theodoros and Liapis, Antonios and Yannakakis, Georgios N},
+  journal={arXiv preprint arXiv:2303.07519},
+  year={2023}
+}
+```
 
+To cite the codebase that trained this model:
+```
 @misc{mesh-transformer-jax,
   author = {Wang, Ben},
   title = {{Mesh-Transformer-JAX: Model-Parallel Implementation of Transformer Language Model with JAX}},
@@ -55,6 +63,7 @@ To cite the codebase that trained this model:
   year = 2021,
   month = May
 }
+```
 
 # Acknowledgements
 This project would not have been possible without compute generously provided by Google through the TPU Research Cloud that generously provided access to Clout TPU VMs used to finetune this model.