update

app.py

@@ -395,84 +395,94 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
         return output_dir, output_dir
 
 
-title = "SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training"
-description = r"""<center><img src='https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/assets/seedvr_logo.png' style='height:40px' alt='SeedVR logo'></center>
-<b>Official Gradio demo</b> for <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'><b>SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training</b></a>.<br>
-🔥 SeedVR2 is a one-step image and video restoration algorithm for real-world and AIGC contents.<br>
-"""
-article = r"""
-If SeedVR series are helpful, please help to ⭐ the <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>Github Repo</a>. Thanks!
-[![GitHub Stars](https://img.shields.io/github/stars/ByteDance-Seed/SeedVR?style=social)](https://github.com/ByteDance-Seed/SeedVR)
-
----
-
-📝 **Notice:** 
-
-To avoid OOM, this demo only support 720p video outputs within 121 frames. For other uses (e.g., image restoration, other resolutions, etc), please refer to the <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>Github Repo</a>.
-
-
-📝 **Limitations:** 
-
-These are the prototype models and the performance may not perfectly align with the paper. Our methods are sometimes not robust to heavy degradations and very large motions, and shares some failure cases with existing methods, e.g., fail to fully remove the degradation or simply generate unpleasing details. Moreover, due to the strong generation ability, Our methods tend to overly generate details on inputs with very light degradations, e.g., 720p AIGC videos, leading to oversharpened results occasionally.
-
-
-📝 **Citation**
-
-If our work is useful for your research, please consider citing:
-```bibtex
-@article{wang2025seedvr2,
-      title={SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training},
-      author={Wang, Jianyi and Lin, Shanchuan and Lin, Zhijie and Ren, Yuxi and Wei, Meng and Yue, Zongsheng and Zhou, Shangchen and Chen, Hao and Zhao, Yang and Yang, Ceyuan and Xiao, Xuefeng and Loy, Chen Change and Jiang, Lu},
-      booktitle={arXiv preprint arXiv:2506.05301},
-      year={2025}
-   }
-
-@inproceedings{wang2025seedvr,
-      title={SeedVR: Seeding Infinity in Diffusion Transformer Towards Generic Video Restoration},
-      author={Wang, Jianyi and Lin, Zhijie and Wei, Meng and Zhao, Yang and Yang, Ceyuan and Loy, Chen Change and Jiang, Lu},
-      booktitle={CVPR},
-      year={2025}
-   }
-```
-
-📋 **License**
-
-This project is licensed under <a rel="license" href="http://www.apache.org/licenses/LICENSE-2.0">Apache License, Version 2.0</a>.
-Redistribution and use for non-commercial purposes should follow this license.
-
-📧 **Contact**
-
-If you have any questions, please feel free to reach me out at <b>[email protected]</b>.
-
-<div>
-🤗 Find Me:
-<a href="https://twitter.com/Iceclearwjy"><img style="margin-top:0.5em; margin-bottom:0.5em" src="https://img.shields.io/twitter/follow/Iceclearwjy?label=%40Iceclearwjy&style=social" alt="Twitter Follow"></a>
-<a href="https://github.com/IceClear"><img style="margin-top:0.5em; margin-bottom:2em" src="https://img.shields.io/github/followers/IceClear?style=social" alt="Github Follow"></a>
-</div>
-
-<center><img src='https://visitor-badge.laobi.icu/badge?page_id=ByteDance-Seed/SeedVR' alt='visitors'></center>
-"""
-
-demo = gr.Interface(
-	generation_loop, [
-		gr.Video(
-            label="Upload a video"
-        ),
-        gr.Number(label="Seeds"),
-        gr.Number(label="fps"),
-	], [
-		gr.Video(label="Output"),
-		gr.File(label="Download the output")
-	],
-	title=title,
-	description=description,
-	article=article,
-	examples=[
-		['./01.mp4', 4, 24],
-		['./02.mp4', 4, 24],
-		['./03.mp4', 4, 24],
-		]
-	)
+with gr.Blocks(title="SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training") as demo:
+    # Top logo and title
+    gr.HTML("""
+        <div style='text-align:center; margin-bottom: 10px;'>
+            <img src='https://huggingface.co/ByteDance-Seed/SeedVR2-3B/resolve/main/assets/seedvr_logo.png' style='height:40px;' alt='SeedVR logo'/>
+        </div>
+        <p><b>Official Gradio demo</b> for 
+        <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>
+        <b>SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training</b></a>.<br>
+        🔥 <b>SeedVR2</b> is a one-step image and video restoration algorithm for real-world and AIGC content.
+        </p>
+    """)
+
+    # Interface
+    with gr.Row():
+        input_video = gr.Video(label="Upload a video")
+        seed = gr.Number(label="Seeds")
+        fps = gr.Number(label="fps")
+
+    with gr.Row():
+        output_video = gr.Video(label="Output")
+        download_link = gr.File(label="Download the output")
+
+    run_button = gr.Button("Run")
+    run_button.click(fn=generation_loop, inputs=[input_video, seed, fps], outputs=[output_video, download_link])
+
+    # Examples
+    gr.Examples(
+        examples=[
+            ["./01.mp4", 4, 24],
+            ["./02.mp4", 4, 24],
+            ["./03.mp4", 4, 24],
+        ],
+        inputs=[input_video, seed, fps]
+    )
+
+    # Article/Footer
+    gr.HTML("""
+        <hr>
+        <p>If you find SeedVR helpful, please ⭐ the 
+        <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>GitHub repository</a>:</p>
+
+        <a href="https://github.com/ByteDance-Seed/SeedVR" target="_blank">
+            <img src="https://img.shields.io/github/stars/ByteDance-Seed/SeedVR?style=social" alt="GitHub Stars">
+        </a>
+
+        <h4>Notice</h4>
+        <p>This demo supports up to <b>720p</b> and <b>121 frames</b>.  
+        For other use cases, check the <a href='https://github.com/ByteDance-Seed/SeedVR' target='_blank'>GitHub repo</a>.</p>
+
+        <h4>Limitations</h4>
+        <p>May fail on heavy degradations or small-motion AIGC clips, causing oversharpening or poor restoration.</p>
+
+        <h4>Citation</h4>
+        <pre style="font-size: 12px;">
+        @article{wang2025seedvr2,
+        title={SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training},
+        author={Wang, Jianyi and ...},
+        booktitle={arXiv preprint arXiv:2506.05301},
+        year={2025}
+        }
+        @inproceedings{wang2025seedvr,
+        title={SeedVR: Seeding Infinity...},
+        booktitle={CVPR},
+        year={2025}
+        }
+        </pre>
+
+        <h4>License</h4>
+        <p>Licensed under the 
+        <a href="http://www.apache.org/licenses/LICENSE-2.0" target="_blank">Apache 2.0 License</a>.</p>
+
+        <h4>Contact</h4>
+        <p>Email: <b>iceclearwjy@gmail.com</b></p>
+
+        <p>
+        <a href="https://twitter.com/Iceclearwjy">
+        <img src="https://img.shields.io/twitter/follow/Iceclearwjy?label=%40Iceclearwjy&style=social" alt="Twitter Follow">
+        </a>
+        <a href="https://github.com/IceClear">
+        <img src="https://img.shields.io/github/followers/IceClear?style=social" alt="GitHub Follow">
+        </a>
+        </p>
+
+        <p style="text-align:center;">
+        <img src="https://visitor-badge.laobi.icu/badge?page_id=ByteDance-Seed/SeedVR" alt="visitors">
+        </p>
+    """)
 
 demo.queue()
 demo.launch()