AVIS-Studio / app.py
forgespecter's picture
Update app.py
3640065 verified
Raw
History Blame Contribute Delete
703 Bytes
import gradio as gr
import torch
from diffusers import StableVideoDiffusionPipeline
from diffusers.utils import export_to_video
# CPU için yapılandırıldı
pipe = StableVideoDiffusionPipeline.from_pretrained(
"stabilityai/stable-video-diffusion-img2vid-xt",
torch_dtype=torch.float32 # CPU'da float32 zorunludur
)
pipe.to("cpu")
def generate_video(image):
image = image.resize((512, 320)) # CPU'da çok büyük görsel hata verir
frames = pipe(image, decode_chunk_size=2, motion_bucket_id=127).frames[0]
export_to_video(frames, "output.mp4", fps=7)
return "output.mp4"
demo = gr.Interface(fn=generate_video, inputs=gr.Image(type="pil"), outputs=gr.Video())
demo.launch()