free-veo3-ai / generate.py
chiraggupta8769's picture
Update generate.py
38b4bdf verified
raw
history blame
513 Bytes
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
import argparse
import shutil
parser = argparse.ArgumentParser()
parser.add_argument("--prompt", required=True)
parser.add_argument("--output", default="modelscope_output.mp4")
args = parser.parse_args()
pipe = pipeline(Tasks.text_to_video_synthesis, model='damo-vilab/modelscope-text-to-video-synthesis')
result = pipe(args.prompt)
# Move video file to expected output name
shutil.move(result["output_video"], args.output)