Update README.md

README.md

@@ -133,7 +133,7 @@ These scores indicate that the model, in its current state, is not a strong ches
 
 You can see a demo game here:
 
-[![StockZero Demo Gameplay Video](https://huggingface.co/nirajandhakal/StockZero/resolve/main/demo_video_thumbnail.png)](https://huggingface.co/nirajandhakal/StockZero/blob/main/v2-gameplay-svg-high-quality.mp4)
+[![StockZero Demo Gameplay Video](https://huggingface.co/nirajandhakal/StockZero-v2/resolve/main/demo_video_thumbnail.png)](https://huggingface.co/nirajandhakal/StockZero-v2/blob/main/v2-gameplay-svg-high-quality.mp4)
 
 
 ## How to Use
@@ -239,4 +239,17 @@ if np.sum(masked_policy_probs) > 0:
 
 print("Policy Output:", masked_policy_probs)
 print("Value Output:", value_output)
-```
+```
+
+### Citation
+
+If you use this model in your research, please cite it as follows:
+
+@misc{stockzero,
+  author = {Nirajan Dhakal},
+  title = {StockZero: A Self-Play Reinforcement Learning Chess Engine},
+  year = {2025},
+  publisher = {Hugging Face},
+  journal = {Hugging Face Model Card},
+  howpublished = {\url{https://huggingface.co/nirajandhakal/StockZero-v2}}
+}