import torch
from diffusers import StableVideoDiffusionPipeline

# CPU पर लोड करने के लिए
pipeline = StableVideoDiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid")  
pipeline.to("cpu", torch.float32)  # ✅ GPU की जगह CPU का इस्तेमाल

# इनफरेंस फंक्शन (वीडियो जनरेट करना)
def generate_video(prompt):
    video = pipeline(prompt).videos  # ✅ अब यह CPU पर रन करेगा
    return video

# टेस्टिंग
prompt = "A cute cat playing with a ball"
output_video = generate_video(prompt)

# आउटपुट सेव करें
output_video[0].save("generated_video.mp4")  # ✅ CPU पर वीडियो सेव होगा