Sabbir / video_pipeline.py
Sabbirr12's picture
Create video_pipeline.py
18f31e3 verified
raw
history blame contribute delete
334 Bytes
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
video_pipe = pipeline(Tasks.text_to_video_synthesis, model='damo/text-to-video-synthesis', device='cuda')
def generate_video(prompt, image_path):
result = video_pipe({'text': prompt, 'input_image': image_path})
return result['video_path']