Update README.md

README.md

@@ -9,12 +9,12 @@ language:
 ---
 
 ## 💡 Model description
-This repo contains a large ocean generative model (OceanGPT) built with ocean science dataset. It should be noted that the OceanGPT is constantly being updated, so the current model is not the final version.
+This repo contains a large language model (OceanGPT) for ocean  science tasks trained with [KnowLM](https://github.com/zjunlp/KnowLM). It should be noted that the OceanGPT is constantly being updated, so the current model is not the final version.
 
 ## 🔍 Intended uses
 You can download the model to generate responses or contact the [email](bizhen_zju@zju.edu.cn) for the online test demo.
 
-## 🛠️ How to use by yourself
+## 🛠️ How to use
 We wil provide several examples soon and you can modify the input according to your needs.
 
 ```python
@@ -27,4 +27,15 @@ We wil provide several examples soon and you can modify the input according to y
 >>> tokenizer = AutoTokenizer.from_pretrained("zjunlp/OceanGPT-7b")
 >>> model = AutoModelForCausalLM.from_pretrained("zjunlp/OceanGPT-7b")
 
-```
+```
+
+## How to cite
+
+```bibtex
+@article{bi2023oceangpt,
+  title={OceanGPT: A Large Language Model for Ocean Science Tasks},
+  author={Bi, Zhen and Zhang, Ningyu and Xue, Yida and Ou, Yixin and Ji, Daxiong and Zheng, Guozhou and Chen, Huajun},
+  journal={arXiv preprint arXiv:2310.02031},
+  year={2023}
+}
+```