add paper link

AI4WA · Jun 6, 2024 · 1ded657 · 1ded657
1 parent a6a3224
commit 1ded657
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -3,6 +3,7 @@
 **Unified Knowledge Graph Construction from Heterogeneous Documents Assisted by Large
 Language Models**
 
+[![Paper](https://img.shields.io/badge/Paper-View-green?style=flat&logo=adobeacrobatreader)](https://arxiv.org/pdf/2406.02962)
 ![PyPI](https://img.shields.io/pypi/v/Docs2KG)
 [![Demo](https://img.shields.io/badge/Demo-Available-blue)](https://docs2kg.ai4wa.com/Video/)
 ![Lint](https://github.com/AI4WA/Docs2KG/actions/workflows/lint.yml/badge.svg)

diff --git a/docs/index.md b/docs/index.md
@@ -3,6 +3,7 @@
 **Unified Knowledge Graph Construction from Heterogeneous Documents Assisted by Large
 Language Models**
 
+[![Paper](https://img.shields.io/badge/Paper-View-green?style=flat&logo=adobeacrobatreader)](https://arxiv.org/pdf/2406.02962)
 ![PyPI](https://img.shields.io/pypi/v/Docs2KG)
 [![Demo](https://img.shields.io/badge/Demo-Available-blue)](https://docs2kg.ai4wa.com/Video/)
 ![Lint](https://github.com/AI4WA/Docs2KG/actions/workflows/lint.yml/badge.svg)
@@ -164,4 +165,19 @@ pip install -r requirements.dev.txt
 pip install -e .
 ```
 
----
+---
+
+## Citation
+
+If you find this package useful, please consider citing our work:
+
+```bibtex
+@misc{sun2024docs2kg,
+      title={Docs2KG: Unified Knowledge Graph Construction from Heterogeneous Documents Assisted by Large Language Models}, 
+      author={Qiang Sun and Yuanyi Luo and Wenxiao Zhang and Sirui Li and Jichunyang Li and Kai Niu and Xiangrui Kong and Wei Liu},
+      year={2024},
+      eprint={2406.02962},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL}
+}
+```