Update README.md

README.md

@@ -122,3 +122,14 @@ inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model.generate(**inputs, max_new_tokens=256)
 
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
+```
+
+## Citation
+```bibtex
+@misc{akinwale2025codet5pytest,
+  author  = {Akinwale, Oluwaferanmi Philip},
+  title   = {CodeT5p-220m PyTest Generator},
+  year    = {2025},
+  note    = {Hugging Face model repository},
+  url     = {https://huggingface.co/Oluwaferanmiii/codet5p-220m-pytest-generator}
+}