Spaces:
Running
Running
| 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) | |