arxiv

app.py

@@ -75,6 +75,22 @@ If ColorFlow is helpful, please help to ⭐ the <a href='https://github.com/Tenc
 📧 **Contact**
 <br>
 If you have any questions, please feel free to reach me out at <b>[email protected]</b>.
+
+📝 **Citation**
+<br>
+If our work is useful for your research, please consider citing:
+```bibtex
+@misc{zhuang2024colorflow,
+      title={ColorFlow: Retrieval-Augmented Image Sequence Colorization}, 
+      author={Junhao Zhuang and Xuan Ju and Zhaoyang Zhang and Yong Liu and Shiyi Zhang and Chun Yuan and Ying Shan},
+      year={2024},
+      eprint={2412.11815},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2412.11815},
+}
+```
+
 """
 
 model_global_path = snapshot_download(repo_id="TencentARC/ColorFlow", cache_dir='./colorflow/', repo_type="model")
@@ -436,7 +452,7 @@ with gr.Blocks() as demo:
     <h3 style="text-align: center; font-size: 1.8em;">Retrieval-Augmented Image Sequence Colorization</h3>
     <p style="text-align: center; font-weight: bold;">
         <a href="https://zhuang2002.github.io/ColorFlow/">Project Page</a> | 
-        <a href="https://arxiv.org/abs/">ArXiv Preprint</a> | 
+        <a href="https://arxiv.org/abs/2412.11815">ArXiv Preprint</a> | 
         <a href="https://github.com/TencentARC/ColorFlow">GitHub Repository</a>
     </p>
     <p style="text-align: center; font-weight: bold;">