Update app.py

app.py

@@ -97,6 +97,14 @@ examples = [
      Image.open("./test_imgs/04.png"), 696, 1024, 3.5, 24, 42]
 ]
 
+header = """
+<div style="text-align: center; display: flex; justify-content: left; gap: 5px;">
+<a href="https://arxiv.org/abs/2505.18445"><img src="https://img.shields.io/badge/ariXv-2505.18445-A42C25.svg" alt="arXiv"></a>
+<a href="https://huggingface.co/showlab/OmniConsistency"><img src="https://img.shields.io/badge/🤗_HuggingFace-Model-ffbd45.svg" alt="HuggingFace"></a>
+<a href="https://github.com/showlab/OmniConsistency"><img src="https://img.shields.io/badge/GitHub-Code-blue.svg?logo=github&" alt="GitHub"></a>
+</div>
+"""
+
 # Gradio interface setup
 def create_gradio_interface():
     lora_names = [
@@ -110,6 +118,7 @@ def create_gradio_interface():
 
     with gr.Blocks() as demo:
         gr.Markdown("# OmniConsistency LoRA Image Generation")
+        gr.Markdown(header, elem_id="header")
         gr.Markdown("Select a LoRA, enter a prompt, and upload an image to generate a new image with OmniConsistency. [View on GitHub](https://github.com/showlab/OmniConsistency)")
         with gr.Row():
             with gr.Column(scale=1):