|
import gradio as gr |
|
from PIL import Image |
|
import numpy as np |
|
|
|
from diffusers import DiffusionPipeline |
|
|
|
|
|
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid-xt") |
|
|
|
|
|
def generate_video(input_image): |
|
|
|
input_image_pil = Image.fromarray((input_image * 255).astype(np.uint8)) |
|
|
|
output_video_path = "output_video.mp4" |
|
pipeline(input_image_pil, output_video_path) |
|
return output_video_path |
|
|
|
|
|
iface = gr.Interface(fn=generate_video, inputs="image", outputs="file") |
|
|
|
|
|
iface.launch(share=True) |