add citation

README.md

@@ -164,4 +164,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}
+}
+```

examples/demo/pdf_exported.py

@@ -35,9 +35,9 @@
     """
 
     # you can name your file here
-    pdf_file = DATA_INPUT_DIR / "tests_pdf" / "4.pdf"
+    pdf_file = DATA_INPUT_DIR / "historic information.pdf"
 
-    output_folder = DATA_OUTPUT_DIR / "4.pdf"
+    output_folder = DATA_OUTPUT_DIR / "historic information.pdf"
     # the output will be default to `DATA_OUTPUT_DIR / "4.pdf" /` folder
     scanned_or_exported = get_scanned_or_exported(pdf_file)
     if scanned_or_exported == PDF_TYPE_SCANNED: