Update app.py

app.py

@@ -66,30 +66,30 @@ from util import util
 from huggingface_hub import snapshot_download
 
 
-article = r"""
-If ColorFlow is helpful, please help to ⭐ the <a href='https://github.com/TencentARC/ColorFlow' target='_blank'>Github Repo</a>. Thanks! [![GitHub Stars](https://img.shields.io/github/stars/TencentARC/ColorFlow)](https://github.com/TencentARC/ColorFlow)
----
-
-📧 **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},
-}
-```
-
-"""
+# article = r"""
+# If ColorFlow is helpful, please help to ⭐ the <a href='https://github.com/TencentARC/ColorFlow' target='_blank'>Github Repo</a>. Thanks! [![GitHub Stars](https://img.shields.io/github/stars/TencentARC/ColorFlow)](https://github.com/TencentARC/ColorFlow)
+# ---
+
+# 📧 **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")
 print(model_global_path)
@@ -538,5 +538,29 @@ with gr.Blocks() as demo:
             examples_per_page=6,
         )
         
-    gr.Markdown(article)
+    gr.HTML(
+        '<a href="https://github.com/TencentARC/ColorFlow"><img src="https://img.shields.io/github/stars/TencentARC/ColorFlow" alt="GitHub Stars"></a>'
+    )
+    gr.HTML(
+        """
+        📧 **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:
+        ```
+        @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},
+        }
+        ```
+        """
+    )
+
+    
 demo.launch()