models/animatdiff.py

from diffusers import AnimateDiffPipeline

pipe = AnimateDiffPipeline.from_pretrained("animatediff")

def generate_frames(prompt):
    result = pipe(prompt, num_frames=16)
    return result.frames