Update app.py

app.py

@@ -126,10 +126,10 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
     total_frames = get_video_dimension(video_path)[3]
 
     # Resize the video
-    resized = resize_video(video_path, 'resized.mp4', 512, 512, target_fps)
+    #resized = resize_video(video_path, 'resized.mp4', 512, 512, target_fps)
 
     # normalize pixels
-    normalized = normalize_and_save_video(resized, 'normalized.mp4')
+    #normalized = normalize_and_save_video(resized, 'normalized.mp4')
 
     output_path = 'output/'
     os.makedirs(output_path, exist_ok=True)
@@ -141,27 +141,27 @@ def run_inference(prompt, video_path, condition, video_length, seed, steps):
 
     print(f"RUNNING INFERENCE ...")
     if video_length > 12:
-        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 --is_long_video"
+        command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --condition '{condition}' --video_path '{video_path}' --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 --is_long_video"
     else:
-        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"
+        command = f"python inference.py --prompt '{prompt}' --inference_steps {steps} --condition '{condition}' --video_path '{video_path}' --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'
     except cuda.Error as e:
         return f"CUDA Error: {e}", None
     except RuntimeError as e:
         return f"Runtime Error: {e}", None
 
+    # 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'
  
 
 css="""