Update README.md

README.md

@@ -52,4 +52,14 @@ sketch = "<mask> Conference on Empirical Methods <mask> submission of research p
 # 3. here we go!
 generated_text = genius(sketch, num_beams=3, do_sample=True, max_length=200)[0]['generated_text']
 print(generated_text)
+```
+
+If you find our paper/code/demo useful, please cite our paper:
+```
+@article{guo2022genius,
+  title={GENIUS: Sketch-based Language Model Pre-training via Extreme and Selective Masking for Text Generation and Augmentation},
+  author={Guo, Biyang and Gong, Yeyun and Shen, Yelong and Han, Songqiao and Huang, Hailiang and Duan, Nan and Chen, Weizhu},
+  journal={arXiv preprint arXiv:2211.10330},
+  year={2022}
+}
 ```