blabla974's picture
Switch app.py to use Stable Video Diffusion (video_generator_svd.py)
dc8f698 verified
import gradio as gr
from motion_utils.video_generator_svd import generate_video_from_image
import tempfile
import os
def process(image, duration, fps):
with tempfile.TemporaryDirectory() as tmpdir:
image_path = os.path.join(tmpdir, "input_image.png")
output_path = os.path.join(tmpdir, "output_video.mp4")
image.save(image_path)
generate_video_from_image(image_path, duration, fps, output_path)
return output_path
with gr.Blocks() as demo:
gr.Markdown("### SkyReels V3 - Stable Video Diffusion (sans prompt)")
with gr.Row():
with gr.Column():
input_image = gr.Image(label="Image source", type="pil")
duration = gr.Slider(1, 10, value=4, step=1, label="Durée (secondes)")
fps = gr.Slider(5, 30, value=10, step=1, label="FPS (images/seconde)")
btn = gr.Button("Générer la vidéo")
with gr.Column():
output_video = gr.Video(label="Vidéo générée")
btn.click(fn=process, inputs=[input_image, duration, fps], outputs=output_video)
demo.launch()