Add application file

app.py

@@ -17,9 +17,9 @@ examples = [
 ]
 
 
-def venhancer_demo(result_dir="./tmp/", version="v1"):
+def venhancer_demo(result_dir="./tmp/"):
     css = """#input_video {max-width: 1024px !important} #output_vid {max-width: 2048px; max-height:1280px}"""
-    venhancer = VEnhancer_sr(result_dir, version=version)
+    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> \
@@ -55,13 +55,13 @@ def venhancer_demo(result_dir="./tmp/", version="v1"):
 
                 gr.Examples(
                     examples=examples,
-                    inputs=[input_video, input_text, up_scale, target_fps, noise_aug],
+                    inputs=[input_video, input_text],
                     outputs=[output_video],
                     fn=venhancer.enhance_a_video,
                     cache_examples=False,
                 )
             end_btn.click(
-                inputs=[input_video, input_text, up_scale, target_fps, noise_aug],
+                inputs=[input_video, input_text],
                 outputs=[output_video],
                 fn=venhancer.enhance_a_video,
             )
@@ -69,20 +69,8 @@ def venhancer_demo(result_dir="./tmp/", version="v1"):
     return venhancer_iface
 
 
-def get_parser():
-    parser = argparse.ArgumentParser()
-    parser.add_argument(
-        "--version", type=str, default="v1", choices=["v1", "v2"], help="select the checkpoint version."
-    )
-
-    return parser
-
-
 if __name__ == "__main__":
-    parser = get_parser()
-    args = parser.parse_args()
-
     result_dir = os.path.join("./", "results")
-    venhancer_iface = venhancer_demo(result_dir, args.version)
+    venhancer_iface = venhancer_demo(result_dir)
     venhancer_iface.queue(max_size=12)
     venhancer_iface.launch(max_threads=1)