Update app.py

app.py

@@ -60,19 +60,22 @@ def get_samples():
         for sample in sample_list
     ]
 
-header = """
-# ByteMorph
-
-<div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
-<a href=""><img src="https://img.shields.io/badge/ariXv-Paper-A42C25.svg" alt="arXiv"></a>
-<a href="https://huggingface.co/datasets/Boese0601/ByteMorph-Bench"><img src="https://img.shields.io/badge/🤗-Model-ffbd45.svg" alt="HuggingFace"></a>
-<a href="https://github.com/Boese0601/ByteMorph"><img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=github&" alt="GitHub"></a>
-</div>
-"""
 
 def create_app():
     with gr.Blocks() as app:
-        gr.Markdown(header, elem_id="header")
+        gr.HTML(
+            """
+            <div style="text-align: center;">
+                <h2>ByteMorpher</h2>
+                <a href="https://boese0601.github.io/bytemorph/" target="_blank"><img src="https://img.shields.io/badge/Project-Website-blue" style="display:inline-block;"></a>
+                <a href="https://github.com/Boese0601/ByteMorph" target="_blank"><img src="https://img.shields.io/github/stars/Boese0601/ByteMorph?label=GitHub%20%E2%98%85&logo=github&color=green" style="display:inline-block;"></a>
+                <a href="https://huggingface.co/datasets/Boese0601/ByteMorph-6M-Demo" target="_blank"><img src="https://img.shields.io/badge/🤗%20Hugging%20Face-Dataset_Demo-yellow" style="display:inline-block;"></a>
+                <a href="https://huggingface.co/datasets/Boese0601/ByteMorph-Bench" target="_blank"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20HuggingFace%20-Benchmark-yellow" style="display:inline-block;"></a>
+                <a href="https://huggingface.co/datasets/Boese0601/ByteMorpher" target="_blank"><img src="https://img.shields.io/badge/🤗%20Hugging%20Face%20-Model-green" style="display:inline-block;"></a>
+            </div>
+            """
+        )
+        # gr.Markdown(header, elem_id="header")
         with gr.Row(equal_height=False):
             with gr.Column(variant="panel", elem_classes="inputPanel"):
                 original_image = gr.Image(