import gradio as gr
import torch
from diffusers import StableVideoDiffusionPipeline
from PIL import Image

# मॉडल लोड करें
pipe = StableVideoDiffusionPipeline.from_pretrained(
    "stabilityai/stable-video-diffusion-img2vid",
    torch_dtype=torch.float16
).to("cpu")  # Hugging Face Space में CPU ही चलेगा

def generate_video(image):
    image = Image.open(image).convert("RGB")  # इमेज लोड करें
    video = pipe(image, num_inference_steps=25).frames
    video_path = "generated_video.mp4"
    video[0].save(video_path)  # वीडियो सेव करें
    return video_path

# Gradio Interface बनाएं
iface = gr.Interface(
    fn=generate_video,
    inputs=gr.Image(type="pil"),
    outputs=gr.Video()
)

iface.launch()