Update README.md

README.md

@@ -8,6 +8,14 @@ license: apache-2.0
 [Github](https://github.com/Lyu6PosHao/ProLLaMA) for more information
 
 ProLLaMA_Stage_1 is based on Llama-2-7b, so please follow the license of Llama2.
+
+# Input format:
+```text
+Seq=
+#You can also specify the first few amino acids of the protein sequence:
+Seq=<MAPGGMPRE
+```
+
 # Quick usage:
 ```bash
 # you can replace the model_path with your local path
@@ -109,9 +117,12 @@ if __name__ == '__main__':
             print("All the outputs have been saved in",args.output_file)
 ```
 
-# Input format:
-```text
-Seq=
-#You can also specify the first few amino acids of the protein sequence:
-Seq=<MAPGGMPRE
+# Citation:
+```
+@article{lv2024prollama,
+  title={ProLLaMA: A Protein Large Language Model for Multi-Task Protein Language Processing},
+  author={Lv, Liuzhenghao and Lin, Zongying and Li, Hao and Liu, Yuyang and Cui, Jiaxi and Chen, Calvin Yu-Chian and Yuan, Li and Tian, Yonghong},
+  journal={arXiv preprint arXiv:2402.16445},
+  year={2024}
+}
 ```