Update README.md

README.md

@@ -8,11 +8,13 @@ tags:
 - sft
 license: mit
 datasets:
-- MacPaw/GUIrilla
+- macpaw-research/GUIrilla-Task
 ---
 
 # GUIrilla-See-7B
 
+[![arXiv](https://img.shields.io/badge/arXiv-2510.16051-b31b1b.svg)](https://arxiv.org/abs/2510.16051)
+
 *Vision–language grounding for graphical user interfaces*
 
 ---
@@ -117,4 +119,15 @@ Trained on [GUIrilla-Task](https://huggingface.co/datasets/MacPaw/GUIrilla).
 
 ## License
 
-MIT (see `LICENSE`).
+MIT (see `LICENSE`).
+
+## Citation
+```
+@article{garkot2025guirilla,
+  title={GUIrilla: A Scalable Framework for Automated Desktop UI Exploration},
+  author={Garkot, Sofiya and Shamrai, Maksym and Synytsia, Ivan and Hirna, Mariya},
+  journal={arXiv preprint arXiv:2510.16051},
+  year={2025},
+  url={https://arxiv.org/abs/2510.16051}
+}
+```