Update README.md

README.md

@@ -27,7 +27,7 @@ The goal of the DocuMint model is to generate docstrings that are concise (brief
 
 <!-- Provide the basic links for the model. -->
 
-- **Repository:** [https://github.com/Docu-Mint/DocuMint]
+- **Repository:** [GitHub](https://github.com/Docu-Mint/DocuMint)
 - **Paper:** [DocuMint: Docstring Generation for Python using Small Language Models]() Link TODO
 
 ## Uses
@@ -99,10 +99,11 @@ Fine-tuning was performed using an Intel 12900K CPU, an Nvidia RTX-3090 GPU, and
 
 (TODO)
 ```
-@misc{poudel2024documint,
-      title={DocuMint: Docstring Generation for Python using Small Language Models}, 
-      author={Bibek Poudel* and Adam Cook* and Sekou Traore* and Shelah Ameli*},
-      year={2024},
+@article{poudel2024documint,
+  title={DocuMint: Docstring Generation for Python using Small Language Models},
+  author={Poudel, Bibek and Cook, Adam and Traore, Sekou and Ameli, Shelah},
+  journal={arXiv preprint arXiv:},
+  year={2024}
 }
 ```