Update app.py

app.py

@@ -247,8 +247,19 @@ def cleanup(state):
 # Build Gradio UI
 demo = gr.Blocks()
 with demo:
-    gr.Markdown("## Image Processing Pipeline (Single Button)")
-    gr.Markdown("Upload an image and click 'Run Pipeline' to execute all steps and download the mesh.")
+    gr.Markdown("## Pixel3dmm [Image Mode]")
+    gr.Markdown("Versatile Screen-Space Priors for Single-Image 3D Face Reconstruction.")
+    gr.HTML(
+        """
+        <br>
+        <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
+            <a href="https://github.com/SimonGiebenhain/pixel3dmm">
+                <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+            </a>
+        </div>
+        """
+    )
+    
     with gr.Row():
         with gr.Column():
             image_in = gr.Image(label="Upload Image", type="numpy", height=512)