zerogpu

app.py

@@ -1,9 +1,6 @@
-import argparse
+import spaces
 import os
-import random
-
 import gradio as gr
-
 from video_super_resolution.scripts.inference_sr import VEnhancer_sr
 
 examples = [
@@ -16,15 +13,21 @@ examples = [
     ["prompts/gwen.mp4", "Gwen Stacy reading a book, black and white", 4, 24, 250],
 ]
 
+# Define a GPU-decorated function for enhancement
+@spaces.GPU(duration=1200)
+def enhance_with_gpu(input_video, input_text):
+    return venhancer.enhance_a_video(input_video, input_text)
 
 def venhancer_demo(result_dir="./tmp/"):
     css = """#input_video {max-width: 1024px !important} #output_vid {max-width: 2048px; max-height:1280px}"""
+    global venhancer
     venhancer = VEnhancer_sr(result_dir)
+    
     with gr.Blocks(analytics_enabled=False, css=css) as venhancer_iface:
         gr.Markdown(
             "<div align='center'> <h1> VEnhancer </span> </h1> \
-                     <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2407.07667'> [ArXiv] </a>\
-                     <a style='font-size:18px;color: #000000' href='https://yhliu04.github.io/STAR/'> [Project Page] </a> \
+                     <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2501.02976'> [ArXiv] </a>\
+                     <a style='font-size:18px;color: #000000' href='https://nju-pcalab.github.io/projects/STAR'> [Project Page] </a> \
                      <a style='font-size:18px;color: #000000' href='https://github.com/NJU-PCALab/STAR'> [Github] </a> </div>"
         )
         with gr.Tab(label="VEnhancer"):
@@ -45,13 +48,13 @@ def venhancer_demo(result_dir="./tmp/"):
                     examples=examples,
                     inputs=[input_video, input_text],
                     outputs=[output_video],
-                    fn=venhancer.enhance_a_video,
+                    fn=enhance_with_gpu,  # Use the GPU-decorated function
                     cache_examples=False,
                 )
             end_btn.click(
                 inputs=[input_video, input_text],
                 outputs=[output_video],
-                fn=venhancer.enhance_a_video,
+                fn=enhance_with_gpu,  # Use the GPU-decorated function
             )
 
     return venhancer_iface
@@ -61,4 +64,4 @@ if __name__ == "__main__":
     result_dir = os.path.join("./", "results")
     venhancer_iface = venhancer_demo(result_dir)
     venhancer_iface.queue(max_size=12)
-    venhancer_iface.launch(max_threads=1)
+    venhancer_iface.launch(max_threads=1)