Update app.py

app.py

@@ -98,21 +98,38 @@ def run_xportrait(source_image, driving_video, seed, uc_scale, best_frame, out_f
         return f"An error occurred: {e}", None
 
 # Set up Gradio interface
-with gr.Blocks() as demo:
+css="""
+div#frames-gallery{
+    overflow: scroll!important;
+}
+"""
+with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
+        gr.Markdown("# X-Portrait: Expressive Portrait Animation with Hierarchical Motion Attention")
+        gr.HTML("""
+        <div style="display:flex;column-gap:4px;">
+            <a href='https://github.com/bytedance/X-Portrait'>
+                <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+            </a> 
+            <a href='https://byteaigc.github.io/x-portrait/'>
+                <img src='https://img.shields.io/badge/Project-Page-green'>
+            </a>
+        </div>
+        """)
         with gr.Row():
             with gr.Column():
-                source_image = gr.Image(label="Source Image", type="filepath")
-                driving_video = gr.Video(label="Driving Video")
                 with gr.Row():
-                    seed = gr.Number(value=999, label="Seed")
-                    uc_scale = gr.Number(value=5, label="UC Scale")
+                    source_image = gr.Image(label="Source Image", type="filepath")
+                    driving_video = gr.Video(label="Driving Video")
                 with gr.Group():
                     with gr.Row():
                         best_frame = gr.Number(value=36, label="Best Frame")
                         out_frames = gr.Number(value=-1, label="Out Frames")
                     with gr.Accordion("Driving video Frames"):
-                        driving_frames = gr.Gallery(show_label=True, columns=6, height=512)
+                        driving_frames = gr.Gallery(show_label=True, columns=6, height=512, elem_id="frames-gallery")
+                with gr.Row():
+                    seed = gr.Number(value=999, label="Seed")
+                    uc_scale = gr.Number(value=5, label="UC Scale")
                 with gr.Row():
                     num_mix = gr.Number(value=4, label="Number of Mix")
                     ddim_steps = gr.Number(value=30, label="DDIM Steps")
@@ -120,6 +137,13 @@ with gr.Blocks() as demo:
             with gr.Column():
                 video_output = gr.Video(label="Output Video")
                 status = gr.Textbox(label="status")
+                gr.Examples(
+                    examples=[
+                        ["./assets/source_image.png", "./assets/driving_video.mp4"]
+                    ],
+                    inputs=[source_image, driving_video]
+                )
+
 
     driving_video.upload(
         fn = extract_frames_with_labels,