Update app.py

app.py

@@ -418,6 +418,15 @@ init_app()
 
 with gr.Blocks(title="Z-Image Demo") as demo:
     gr.Markdown("# Z-Image Generation Demo")
+    gr.Markdown(
+        """
+        <div style="text-align: center; margin: 10px 0;">
+            <a href="https://github.com/Tongyi-MAI/Z-Image" target="_blank">
+                <img src="https://img.shields.io/badge/github-repo-blue?logo=github" alt="GitHub Repo">
+            </a>
+        </div>
+        """
+    )
 
     with gr.Row():
         with gr.Column(scale=1):
@@ -472,7 +481,8 @@ with gr.Blocks(title="Z-Image Demo") as demo:
 
     # Dummy enable_enhance variable set to False
     enable_enhance = gr.State(value=False)
-    
+
+    # Update seed function
     def update_seed(current_seed, random_seed_enabled):
         if random_seed_enabled:
             new_seed = random.randint(1, 1000000)