Update app.py

app.py

@@ -146,20 +146,23 @@ with gr.Blocks() as app:
     gr.HTML(
         """
         <div style="display: flex; gap: 10px; justify-content: left;">
-            <a href="https://research.google/blog/a-return-to-hand-written-notes-by-learning-to-read-write/">
-                <img src="https://img.shields.io/badge/Google_Research_Blog-333333?&logo=google&logoColor=white" alt="Google Research Blog">
+            <a href="https://openreview.net/forum?id=pSyUfV5BqA">
+                <img src="https://img.shields.io/badge/Paper-TMLR-red.svg" alt="Paper">
             </a>
             <a href="https://arxiv.org/abs/2402.05804">
-                <img src="https://img.shields.io/badge/Read_the_Paper-4CAF50?&logo=arxiv&logoColor=white" alt="Read the Paper">
-            </a> 
+                <img src="https://img.shields.io/badge/arXiv-2402.05804-b31b1b.svg" alt="arXiv">
+            </a>
             <a href="https://charlieleee.github.io/publication/inksight/">
-                <img src="https://img.shields.io/badge/🔗_Project_Page-FFA500?&logo=link&logoColor=white" alt="Project Page">
+                <img src="https://img.shields.io/badge/Project-Page-purple.svg" alt="Project Page">
+            </a>
+            <a href="https://huggingface.co/spaces/Derendering/Model-Output-Playground">
+                <img src="https://img.shields.io/badge/Demo-HuggingFace-yellow.svg" alt="Demo">
             </a>
-            <a href="https://huggingface.co/datasets/Derendering/InkSight-Derenderings">
-                <img src="https://img.shields.io/badge/Dataset-InkSight-40AF40?&logo=huggingface&logoColor=white" alt="Hugging Face Dataset">
+            <a href="https://githubtocolab.com/google-research/inksight/blob/main/colab.ipynb">
+                <img src="https://img.shields.io/badge/Colab-Example-blue.svg" alt="Colab">
             </a>
-            <a href="https://github.com/google-research/inksight">
-                <img src="https://img.shields.io/badge/GitHub-InkSight-green?&logo=github&logoColor=white" alt="GitHub Repository">
+            <a href="https://research.google/blog/a-return-to-hand-written-notes-by-learning-to-read-write/">
+                <img src="https://img.shields.io/badge/Google-Research%20Blog-4285F4.svg" alt="Google Research Blog">
             </a>
         </div>
         """
@@ -286,5 +289,20 @@ with gr.Blocks() as app:
 
     gr.HTML(full_svg_display)
 
+    gr.Markdown("## Citation")
+    gr.Code(
+        """@article{
+    mitrevski2025inksight,
+    title={InkSight: Offline-to-Online Handwriting Conversion by Teaching Vision-Language Models to Read and Write},
+    author={Blagoj Mitrevski and Arina Rak and Julian Schnitzler and Chengkun Li and Andrii Maksai and Jesse Berent and Claudiu Cristian Musat},
+    journal={Transactions on Machine Learning Research},
+    issn={2835-8856},
+    year={2025},
+    url={https://openreview.net/forum?id=pSyUfV5BqA},
+    note={}
+}""",
+        label="BibTeX"
+    )
+
 
 app.launch()