Update README.md

README.md

@@ -96,4 +96,16 @@ chunks, token_pos=chunk_text(model,text, tokenizer, prob_threshold=0.5)
 for i, (c,t) in enumerate(zip(chunks,token_pos)):
     print(f'-----chunk: {i}----token_idx: {t}--------')
     print(c)
+```
+## Citation
+
+If this work is helpful, please kindly cite as:
+
+```bibtex
+@article{bert-chunker,
+  title={bert-chunker: Efficient and Trained Chunking for Unstructured Documents}, 
+  author={Yannan Luo},
+  year={2024},
+  url={https://github.com/jackfsuia/bert-chunker}
+}
 ```