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 [deepseek-ai/deepseek-llm-7b-chat](https://huggingface.co/deepseek-ai/deepseek-llm-7b-chat) on the data dataset.
+This model is a fine-tuned version of [deepseek-ai/deepseek-llm-7b-chat](https://huggingface.co/deepseek-ai/deepseek-llm-7b-chat) on our dataset.
+
+For more details, please refer to https://github.com/qiuhuachuan/interactive-agents
 
 ## Model inference
 
@@ -76,10 +78,21 @@ if __name__ == '__main__':
 
 Available for non-commercial use
 
+## Citation
 
-## Training and evaluation data
+If you find our work useful for your research and applications, please cite using this BibTeX:
 
-https://github.com/qiuhuachuan/interactive-agents
+```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