Add arxiv link

README.md

@@ -22,4 +22,4 @@ Official demo of 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]()] [[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)] [[pi-Qwen Demo🤗](https://huggingface.co/spaces/Lakonik/pi-Qwen)] [[pi-FLUX Demo🤗](https://huggingface.co/spaces/Lakonik/pi-FLUX.1)] 

app.py

@@ -48,7 +48,7 @@ with gr.Blocks(analytics_enabled=False,
                ) as demo:
 
     md_txt = '# pi-Qwen Demo\n\n' \
-             'Official demo of the paper [pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation](). ' \
+             'Official demo of the paper [pi-Flow: Policy-Based Few-Step Generation via Imitation Distillation](https://arxiv.org/abs/2510.14974). ' \
              '**Base model:** [Qwen-Image](https://huggingface.co/Qwen/Qwen-Image). **Fast policy:** GMFlow. **Code:** [https://github.com/Lakonik/piFlow](https://github.com/Lakonik/piFlow).'
     gr.Markdown(md_txt)
 

lakonlab/pipelines/piflux_pipeline.py

@@ -74,7 +74,7 @@ class PiFluxPipeline(FluxPipeline, PiFlowLoaderMixin):
     r"""
     The policy-based Flux pipeline for text-to-image generation.
 
-    Reference: Todo: add paper link
+    Reference: https://arxiv.org/abs/2510.14974
 
     Args:
         transformer ([`FluxTransformer2DModel`]):

lakonlab/pipelines/piqwen_pipeline.py

@@ -66,7 +66,7 @@ class PiQwenImagePipeline(QwenImagePipeline, PiFlowLoaderMixin):
     r"""
     The policy-based QwenImage pipeline for text-to-image generation.
 
-    Reference: Todo: add paper link
+    Reference: https://arxiv.org/abs/2510.14974
 
     Args:
         transformer ([`QwenImageTransformer2DModel`]):