File size: 653 Bytes
0f2dddc e2436ee 0f2dddc e2436ee 0f2dddc e2436ee 0f2dddc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from diffusers import StableDiffusionPipeline
import os
import ffmpeg
model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
def generate_video(prompt):
frames = []
for i in range(10): # Generate 10 frames
image = model(prompt).images[0]
frame_path = f"frame_{i}.png"
image.save(frame_path)
frames.append(frame_path)
output_video = "output.mp4"
(
ffmpeg
.input("frame_%d.png", framerate=1)
.output(output_video)
.run(overwrite_output=True)
)
# Clean up frames
for frame in frames:
os.remove(frame)
return output_video |