Update README.md

README.md

@@ -13,9 +13,11 @@ model-index:
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# sft
+# Details
 
-This model is a fine-tuned version of [Qwen/Qwen2-7B-Instruct](https://huggingface.co/Qwen/Qwen2-7B-Instruct) on the data dataset.
+This model is a fine-tuned version of [Qwen/Qwen2-7B-Instruct](https://huggingface.co/Qwen/Qwen2-7B-Instruct) on our dataset.
+
+**For more details, please refer to https://github.com/qiuhuachuan/interactive-agents**
 
 ## Model inference
 
@@ -94,9 +96,22 @@ if __name__ == '__main__':
 
 Available for non-commercial use
 
-## Training and evaluation data
 
-https://github.com/qiuhuachuan/interactive-agents
+## Citation
+
+If you find our work useful for your research and applications, please cite using this BibTeX:
+
+```bibtex
+@misc{qiu2024interactiveagents,
+      title={Interactive Agents: Simulating Counselor-Client Psychological Counseling via Role-Playing LLM-to-LLM Interactions},
+      author={Huachuan Qiu and Zhenzhong Lan},
+      year={2024},
+      eprint={2408.15787},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2408.15787},
+}
+```
 
 ## Training procedure