Update README.md

README.md

@@ -7,13 +7,19 @@ widget:
 - text: 'find all files in directory "/mydir" with extension ".txt"'
 ---
 
-# MarianCG: A TRANSFORMER MODEL FOR AUTOMATIC CODE GENERATION
-This model is to improve the solving of the code generation problem and implement a transformer model that can work with high accurate results. We implemented MarianCG transformer model which is a code generation model that can be able to generate code from natural language. This work declares the impact of using Marian machine translation model for solving the problem of code generation. In our implementation we prove that a machine translation model can be operated and working as a code generation model. Finally, we set the new contributors and state-of-the-art on CoNaLa reaching 10.2 Exact Match Accuracy and a BLEU score of 34.43 in the code generation problem with CoNaLa dataset.
+```
+```
+[![PWC](https://img.shields.io/endpoint.svg?url=https://paperswithcode.com/badge/mariancg-a-code-generation-transformer-model/code-generation-on-conala)](https://paperswithcode.com/sota/code-generation-on-conala?p=mariancg-a-code-generation-transformer-model)
+```
+
+```
+
+# MarianCG: a code generation transformer model inspired by machine translation
+This model is to improve the solving of the code generation problem and implement a transformer model that can work with high accurate results. We implemented MarianCG transformer model which is a code generation model that can be able to generate code from natural language. This work declares the impact of using Marian machine translation model for solving the problem of code generation. In our implementation, we prove that a machine translation model can be operated and working as a code generation model. Finally, we set the new contributors and state-of-the-art on CoNaLa reaching a BLEU score of 30.92 and Exact Match Accuracy of 6.2 in the code generation problem with CoNaLa dataset.
 
 MarianCG model and its implemetation with the code of training and the generated output is available at this repository:
 https://github.com/AhmedSSoliman/MarianCG-NL-to-Code
 
-
 CoNaLa Dataset for Code Generation is available at
 https://huggingface.co/datasets/AhmedSSoliman/CoNaLa-Large
 
@@ -39,4 +45,25 @@ Tasks:
 - Code Generation
 - Text2Text Generation
 - Text Generation
----
+---
+
+
+
+
+# Citation
+
+We now have a paper for this work and you can cite:
+
+```
+@article{soliman2022mariancg,
+  title={MarianCG: a code generation transformer model inspired by machine translation},
+  author={Soliman, Ahmed S and Hadhoud, Mayada M and Shaheen, Samir I},
+  journal={Journal of Engineering and Applied Science},
+  volume={69},
+  number={1},
+  pages={1--23},
+  year={2022},
+  publisher={SpringerOpen}
+}
+
+```