Imag / app.py
Baraaqasem's picture
Update app.py
8a6355e verified
raw
history blame
247 Bytes
import videogen_hub
model = videogen_hub.load('VideoCrafter2')
video = model.infer_one_video(prompt="A child excitedly swings on a rusty swing set, laughter filling the air.")
# Here video is a torch tensor of shape torch.Size([16, 3, 320, 512])