Update app.py

app.py

@@ -111,7 +111,7 @@ def normalize_and_save_video(input_video_path, output_video_path):
 
     
 
-def run_inference(prompt, video_path, condition, video_length, seed):
+def run_inference(prompt, video_path, condition, video_length, seed, steps):
     
     seed = math.floor(seed)
     
@@ -141,21 +141,24 @@ def run_inference(prompt, video_path, condition, video_length, seed):
 
     print(f"RUNNING INFERENCE ...")
     if video_length > 12:
-        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result' --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 10 12 --is_long_video"
+        command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result' --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 10 12 --is_long_video"
     else:
-        command = f"python inference.py --prompt '{prompt}' --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result'  --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20"
-    subprocess.run(command, shell=True)
+        command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --condition '{condition}' --video_path '{normalized}' --output_path '{output_path}' --temp_chunk_path 'result'  --width 512 --height 512 --fps {target_fps} --seed {seed} --video_length {video_length} --smoother_steps 19 20"
+    try:
+        subprocess.run(command, shell=True)
     
-    # Construct the video path
-    video_path_output = os.path.join(output_path, f"result.mp4")
-
-    # Resize to original video input size
-    o_width = get_video_dimension(video_path)[0]
-    o_height = get_video_dimension(video_path)[1]
-    resize_video(video_path_output, 'resized_final.mp4', o_width, o_height, target_fps)
-
-    print(f"FINISHED !")
-    return "done", 'resized_final.mp4'
+        # Construct the video path
+        video_path_output = os.path.join(output_path, f"result.mp4")
+    
+        # Resize to original video input size
+        o_width = get_video_dimension(video_path)[0]
+        o_height = get_video_dimension(video_path)[1]
+        resize_video(video_path_output, 'resized_final.mp4', o_width, o_height, target_fps)
+    
+        print(f"FINISHED !")
+        return "done", 'resized_final.mp4'
+    except:
+        return "error", None
 
  
 
@@ -177,6 +180,7 @@ with gr.Blocks(css=css) as demo:
                     with gr.Row():
                         condition = gr.Dropdown(label="Condition", choices=["depth", "canny", "pose"], value="depth")
                         seed = gr.Number(label="seed", value=42)
+                    inference_steps = gr.Slider(label="Inference steps", minimum=12, maximum=50, step=1, value=25)
                 submit_btn = gr.Button("Submit")
             with gr.Column():
                 video_res = gr.Video(label="result")
@@ -191,7 +195,8 @@ with gr.Blocks(css=css) as demo:
                              video_path,
                              condition,
                              video_length,
-                             seed
+                             seed,
+                             inference_steps
                             ],
                     outputs=[status, video_res])