Update app.py

app.py

@@ -3,6 +3,7 @@ import cv2
 import numpy as np
 import torch
 from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
+from moviepy.editor import ImageSequenceClip
 
 pipe = DiffusionPipeline.from_pretrained("cerspense/zeroscope_v2_576w", torch_dtype=torch.float16)
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
@@ -13,21 +14,15 @@ def export_to_video(video_frames):
     video_frames = np.array(video_frames)
     video_frames = (video_frames * 255).astype(np.uint8)
     
-    # Get the dimensions of the frames
-    height, width, channels = video_frames.shape[2:]
+    # Convert frames from (N, H, W, C) to list of (H, W, C)
+    frames_list = [frame for frame in video_frames]
     
-    # Define the video writer object
-    fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # Codec for .mp4 files
+    # Create a video clip from the frames
     fps = 24  # Frames per second
-    video_writer = cv2.VideoWriter('output_video.mp4', fourcc, fps, (width, height))
+    clip = ImageSequenceClip(frames_list, fps=fps)
     
-    # Write each frame to the video
-    for i in range(video_frames.shape[0]):
-        frame = video_frames[i]
-        video_writer.write(frame)
-    
-    # Release the video writer object
-    video_writer.release()
+    # Write the video file
+    clip.write_videofile("output_video.mp4", codec="libx264")
     
     print("Video has been created successfully.")
     return 'output_video.mp4'