Update README.md

README.md

@@ -6,8 +6,8 @@ language:
 - en
 pipeline_tag: text-generation
 ---
-# Description
-The first model for automatic text-to-video prompt generation. It is finetuned on [VidProM](https://huggingface.co/datasets/WenhaoWang/VidProM) using [Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1).
+
+# The first model for automatic text-to-video prompt generation. It is fine-tuned on [VidProM](https://huggingface.co/datasets/WenhaoWang/VidProM) using [Mistral-7B-v0.1](https://huggingface.co/mistralai/Mistral-7B-v0.1).
 
 # Usage
 
@@ -54,4 +54,21 @@ An undemwater world.. A young woman, SARA, is sitting at a table, looking distre
 An undemwater world, . the man is talking to other people.
 An undemwater world, a vibrant coral reef teeming with mine life and a symphony of colors that thrived beneath the waves..
 An undemwater world, 1968 film, 32k resolution, shp, captured by Phantom High-Speed Camera, dynamic dramatic lighting, highly detailed cinematic chacters, hyper-detailed, insan.
-```
+```
+
+# License
+
+The model is licensed under the [CC BY-NC 4.0 license](https://creativecommons.org/licenses/by-nc/4.0/deed.en).
+
+# Citation
+
+@article{wang2024vidprom,
+  title={VidProM: A Million-scale Real Prompt-Gallery Dataset for Text-to-Video Diffusion Models},
+  author={Wang, Wenhao and Yang, Yi},
+  journal={arXiv preprint arXiv:2403.06098},
+  year={2024}
+}
+
+# Contact
+
+If you have any questions, feel free to contact Wenhao Wang (wangwenhao0716@gmail.com).