Spaces:
Running
Running
import gradio as gr | |
import torch | |
from diffusers import DiffusionPipeline | |
model_id = "cerspense/zeroscope_v2_576w" | |
pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16) | |
pipe.to("cuda" if torch.cuda.is_available() else "cpu") | |
def text_to_video(prompt): | |
output = pipe(prompt, num_inference_steps=25) | |
video_frames = output.frames[0] | |
output_path = "output.mp4" | |
import imageio | |
imageio.mimsave(output_path, video_frames, fps=8) | |
return output_path | |
gr.Interface( | |
fn=text_to_video, | |
inputs=gr.Textbox(label="أدخل وصف الفيديو (بالإنجليزية)"), | |
outputs=gr.Video(label="الفيديو الناتج"), | |
title="تحويل نص إلى فيديو باستخدام ZeroScope" | |
).launch() | |