fix

app.py

@@ -138,16 +138,16 @@ def do_sample(
                 .astype(np.uint8)
             )
             # write_video(video_path, frames, fps=6)
-            writer = cv2.VideoWriter(
-                video_path,
-                cv2.VideoWriter_fourcc("m", "p", "4", "v"),
-                6,
-                (frames.shape[-1], frames.shape[-2]),
-            )
-            for fr in frames:
-                writer.write(cv2.cvtColor(fr, cv2.COLOR_RGB2BGR))
-            writer.release()
-            # imageio.mimwrite(video_path, frames)
+            # writer = cv2.VideoWriter(
+            #     video_path,
+            #     cv2.VideoWriter_fourcc("m", "p", "4", "v"),
+            #     6,
+            #     (frames.shape[-1], frames.shape[-2]),
+            # )
+            # for fr in frames:
+            #     writer.write(cv2.cvtColor(fr, cv2.COLOR_RGB2BGR))
+            # writer.release()
+            imageio.mimwrite(video_path, frames, fps=6)
 
     return video_path
 

requirements.txt

@@ -44,3 +44,5 @@ wget
 rembg
 kiui
 spaces
+imageio==2.19.3
+imageio-ffmpeg==0.4.7