Update app.py

app.py

@@ -1,4 +1,5 @@
-ort os
+import gradio
+import os
 import numpy as np
 import argparse
 import imageio
@@ -60,14 +61,19 @@ def get_args():
     args = parser.parse_args()
     return args
 
-def infer(prompt, video_path, output_path, condition, video_length, height, width, smoother_steps, is_long_video, seed):
+def infer(prompt, video_path, condition, video_length, is_long_video):
     #args = get_args()
     #os.makedirs(args.output_path, exist_ok=True)
     
     # Height and width should be a multiple of 32
+    output_path = ""
+    height, width = 512
     height = (height // 32) * 32    
     width = (width // 32) * 32    
-
+    smoother_steps = [19, 20]
+    is_long_video = False
+    seed = 42
+   
     if condition == "pose":
         pretrained_model_or_path = "lllyasviel/ControlNet"
         body_model_path = hf_hub_download(pretrained_model_or_path, "annotator/ckpts/body_pose_model.pth", cache_dir="checkpoints")
@@ -132,4 +138,23 @@ def infer(prompt, video_path, output_path, condition, video_length, height, widt
                 ).videos
     save_videos_grid(sample, f"{output_path}/{prompt}.mp4")
 
-    return f"{output_path}/{prompt}.mp4"
+    return f"{output_path}/{prompt}.mp4"
+
+with gr.Blocks() as demo:
+    with gr.Column():
+        prompt = gr.Textbox(label="prompt")
+        video_path = gr.Video(source="upload", type="filepath")
+        condition = gr.Textbox(label="Condition", value="depth")
+        video_length = gr.Slider(label="video length", minimum=1, maximum=15, step=1, value=2)
+        seed = gr.Number(label="seed", valie=42)
+        submit_btn = gr.Button("Submit")
+        video_res = gr.Video(label="result")
+
+    submit_btn.click(fn=infer, 
+                     inputs=[prompt,
+                                      video_path,
+                                      condition,
+                                      video_length,
+                                      seed,
+                                      ]
+                    outputs=[video_res])