vidcraft / app.py
ihsanvp's picture
initial
2c1f0c3
raw
history blame
255 Bytes
import gradio as gr
import torch
import torchvision
def generate(prompt: str):
video = torch.rand(240, 512, 512, 3)
torchvision.io.write_video(video, "video.mp4", fps=24)
gr.Interface(
fn=generate,
inputs="text",
outputs=gr.Video()
)