Update app.py

app.py

@@ -6,6 +6,7 @@ from rembg import remove as rm
 import cv2
 import uuid
 from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
+import moviepy.video.io.ImageSequenceClip
 
 uid=uuid.uuid4()
 if not os.path.exists(f'{uid}-frames'): os.makedirs(f'{uid}-frames')
@@ -60,11 +61,15 @@ def im_2_vid(images,fps):
     width= this_im.shape[1]
     print (width,height)
     size = (width, height)
-    codec = cv2.VideoWriter_fourcc(*'mp4v') #DIVX, DIVD
-    video = cv2.VideoWriter("tmp_vid.mp4", codec, fps, size)
-    for img in images:
-            video.write(img)
-    return (video)
+    movie_clip = moviepy.video.io.ImageSequenceClip.ImageSequenceClip(images, fps)
+    movie_clip.write_videofile(video_name+output_format)    
+    #codec = cv2.VideoWriter_fourcc(*'mp4v') #DIVX, DIVD
+    #video = cv2.VideoWriter("tmp_vid.mp4", codec, fps, size)
+    #for img in images:
+    #        video.write(img)
+    
+    
+    return (movie_clip)
 def rem_bg(vid):
     new_video_in = str(vid)
     capture = cv2.VideoCapture(new_video_in)