Update README.md

README.md

@@ -19,15 +19,19 @@ Distilled 1-step and 2-step ImageNet DiTs proposed in the paper:
 [Sai Bi](https://sai-bi.github.io/)<sup>2</sup><br>
 <sup>1</sup>Stanford University, <sup>2</sup>Adobe Research
 <br>
-[[arXiv]()] [[Code](https://github.com/Lakonik/piFlow)] [[pi-Qwen Demo🤗](https://huggingface.co/spaces/Lakonik/pi-Qwen)] [[pi-FLUX Demo🤗](https://huggingface.co/spaces/Lakonik/pi-FLUX.1)] 
+[[arXiv](https://arxiv.org/abs/2510.14974)] [[Code](https://github.com/Lakonik/piFlow)] [[pi-Qwen Demo🤗](https://huggingface.co/spaces/Lakonik/pi-Qwen)] [[pi-FLUX Demo🤗](https://huggingface.co/spaces/Lakonik/pi-FLUX.1)] 
 
 ![teaser](https://cdn-uploads.huggingface.co/production/uploads/638067fcb334960c987fbeda/H0J1LYUcSS5YqOwZqQ0Jb.jpeg)
 
 ## Citation
 ```
 @misc{piflow,
-  title={pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation},
-  author={Hansheng Chen and Kai Zhang and Hao Tan and Leonidas Guibas and Gordon Wetzstein and Sai Bi},
-  year={2025},
+      title={pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation}, 
+      author={Hansheng Chen and Kai Zhang and Hao Tan and Leonidas Guibas and Gordon Wetzstein and Sai Bi},
+      year={2025},
+      eprint={2510.14974},
+      archivePrefix={arXiv},
+      primaryClass={cs.LG},
+      url={https://arxiv.org/abs/2510.14974}, 
 }
 ```