zerogpu

app.py

@@ -1,7 +1,7 @@
 import spaces
 import os
 import gradio as gr
-from video_super_resolution.scripts.inference_sr import VEnhancer_sr
+from video_super_resolution.scripts.inference_sr import STAR_sr
 
 examples = [
     ["examples/023_klingai_reedit.mp4", "The video shows a panda strumming a guitar on a rock by a tranquil lake at sunset. With its black-and-white fur, the panda sits against a backdrop of mountains and a vibrant sky painted in orange and pink hues. The serene scene highlights relaxation and whimsy, with the panda, guitar, and lake harmoniously positioned. The natural landscape's depth and perspective enhance the focus on the panda's peaceful interaction with the guitar.", 4, 24, 250],
@@ -12,14 +12,14 @@ examples = [
 # Define a GPU-decorated function for enhancement
 @spaces.GPU(duration=120)
 def enhance_with_gpu(input_video, input_text):
-    return venhancer.enhance_a_video(input_video, input_text)
+    return star.enhance_a_video(input_video, input_text)
 
 def star_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)
+    global star
+    star = STAR_sr(result_dir)
     
-    with gr.Blocks(analytics_enabled=False, css=css) as venhancer_iface:
+    with gr.Blocks(analytics_enabled=False, css=css) as star_iface:
         gr.Markdown(
             "<div align='center'> <h1> STAR: Spatial-Temporal Augmentation with Text-to-Video Models for Real-World Video Super-Resolution </span> </h1> \
                      <a style='font-size:18px;color: #000000' href='https://arxiv.org/abs/2501.02976'> [ArXiv] </a>\
@@ -53,11 +53,11 @@ def star_demo(result_dir="./tmp/"):
                 fn=enhance_with_gpu,  # Use the GPU-decorated function
             )
 
-    return venhancer_iface
+    return star_iface
 
 
 if __name__ == "__main__":
     result_dir = os.path.join("./", "results")
-    venhancer_iface = star_demo(result_dir)
-    venhancer_iface.queue(max_size=12)
-    venhancer_iface.launch(max_threads=1)
+    star_iface = star_demo(result_dir)
+    star_iface.queue(max_size=12)
+    star_iface.launch(max_threads=1)

video_super_resolution/scripts/inference_sr.py

@@ -15,7 +15,7 @@ from inference_utils import *
 logger = get_logger()
 
 
-class VEnhancer_sr():
+class STAR_sr():
     def __init__(self, 
                  result_dir='./results/',
                  file_name='000_video.mp4',
@@ -121,7 +121,7 @@ def main():
 
     assert solver_mode in ('fast', 'normal')
 
-    venhancer_sr = VEnhancer_sr(
+    star_sr = STAR_sr(
                             result_dir=save_dir,
                             file_name=file_name,  # new added
                             model_path=model_path,
@@ -133,7 +133,7 @@ def main():
                             variant_info=None,
                             )
 
-    venhancer_sr.enhance_a_video(input_path, prompt)
+    star_sr.enhance_a_video(input_path, prompt)
 
 
 if __name__ == '__main__':