Add application file

app.py

@@ -4,7 +4,7 @@ import random
 
 import gradio as gr
 
-from enhance_a_video import VEnhancer
+from video_super_resolution.scripts.inference_sr import VEnhancer_sr
 
 examples = [
     ["prompts/astronaut.mp4", "An astronaut flying in space, featuring a steady and smooth perspective", 4, 24, 250],
@@ -19,7 +19,7 @@ examples = [
 
 def venhancer_demo(result_dir="./tmp/", version="v1"):
     css = """#input_video {max-width: 1024px !important} #output_vid {max-width: 2048px; max-height:1280px}"""
-    venhancer = VEnhancer(result_dir, version=version)
+    venhancer = VEnhancer_sr(result_dir, version=version)
     with gr.Blocks(analytics_enabled=False, css=css) as venhancer_iface:
         gr.Markdown(
             "<div align='center'> <h1> VEnhancer </span> </h1> \