Update gradio_app.py

gradio_app.py

@@ -95,21 +95,41 @@ def infer(input_video, input_mask):
 
     torch.cuda.empty_cache()
 
-    return output_path, priori_path
+    return output_path
 
 with gr.Blocks() as demo:
 
     with gr.Column():
-        gr.Markdown("# DiffuEraser")
+        gr.Markdown("# DiffuEraser: A Diffusion Model for Video Inpainting")
+        gr.Markdown("DiffuEraser is a diffusion model for video inpainting, which outperforms state-of-the-art model Propainter in both content completeness and temporal consistency while maintaining acceptable efficiency.")
+        gr.HTML("""
+        <div style="display:flex;column-gap:4px;">
+            <a href="https://github.com/lixiaowen-xw/DiffuEraser">
+                <img src='https://img.shields.io/badge/GitHub-Repo-blue'>
+            </a> 
+            <a href="https://lixiaowen-xw.github.io/DiffuEraser-page">
+                <img src='https://img.shields.io/badge/Project-Page-green'>
+            </a>
+            <a href="https://lixiaowen-xw.github.io/DiffuEraser-page">
+                <img src='https://img.shields.io/badge/ArXiv-Paper-red'>
+            </a>
+            <a href="https://huggingface.co/spaces/fffiloni/DiffuEraser-demo?duplicate=true">
+                <img src="https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-sm.svg" alt="Duplicate this Space">
+            </a>
+        </div>
+        """)
 
         with gr.Row():
 
             with gr.Column():
 
-                input_video = gr.Video(label="Input Video (MP4 ONLY")
-                input_mask = gr.Video(label="Input Mask Video (MP4 ONLY")
+                input_video = gr.Video(label="Input Video (MP4 ONLY)")
+                input_mask = gr.Video(label="Input Mask Video (MP4 ONLY)")
                 submit_btn = gr.Button("Submit")
 
+            with gr.Column():
+
+                video_result = gr.Video(label="Result")
                 gr.Examples(
                     examples = [
                         ["./examples/example1/video.mp4", "./examples/example1/mask.mp4"],
@@ -119,16 +139,11 @@ with gr.Blocks() as demo:
                     inputs = [input_video, input_mask]
                 )
 
-            with gr.Column():
-
-                video_result = gr.Video(label="Result")
-                priori_result = gr.Video(label="ProPainter pass")
-
 
     submit_btn.click(
         fn = infer,
         inputs = [input_video, input_mask],
-        outputs = [video_result, priori_result]
+        outputs = [video_result]
     )
 
 demo.queue().launch(show_api=False, show_error=True)