Fix the space: Video component needs a path, not frames (#2)

- Fix the space: Video component needs a path, not frames (7acee682b334bff8a52cf0530dfff937a4e812af)


Co-authored-by: Fabrice TIERCELIN <Fabrice-TIERCELIN@users.noreply.huggingface.co>

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
+from diffusers.utils import export_to_video
 import torch
+import os
 from PIL import Image
 import spaces
 
@@ -26,11 +28,17 @@ def generate_video(image):
       image: A PIL Image object representing the input image.
 
   Returns:
-      A list of PIL Images representing the video frames.
+      The path of a video file.
   """
   video_frames = pipeline(image=image, num_inference_steps=20).images
 
-  return video_frames
+  # Frames to Video
+  os.makedirs("outputs", exist_ok=True)
+  base_count = len(glob(os.path.join("outputs", "*.mp4")))
+  video_path = os.path.join("outputs", f"{base_count:06d}.mp4")
+  export_to_video(video_frames, video_path, fps=6)
+
+  return video_path
 
 # Launch the Gradio app
 interface.launch()