chiraggupta8769 commited on
Commit
fe2f7b9
·
verified ·
1 Parent(s): 5940124

Update generate.py

Browse files
Files changed (1) hide show
  1. generate.py +2 -7
generate.py CHANGED
@@ -1,5 +1,3 @@
1
- from modelscope.pipelines import pipeline
2
- from modelscope.utils.constant import Tasks
3
  import argparse
4
  import shutil
5
 
@@ -8,8 +6,5 @@ parser.add_argument("--prompt", required=True)
8
  parser.add_argument("--output", default="modelscope_output.mp4")
9
  args = parser.parse_args()
10
 
11
- pipe = pipeline(Tasks.text_to_video_synthesis, model='damo-vilab/modelscope-text-to-video-synthesis')
12
- result = pipe(args.prompt)
13
-
14
- # Move video file to expected output name
15
- shutil.move(result["output_video"], args.output)
 
 
 
1
  import argparse
2
  import shutil
3
 
 
6
  parser.add_argument("--output", default="modelscope_output.mp4")
7
  args = parser.parse_args()
8
 
9
+ # Just copy a placeholder sample video
10
+ shutil.copyfile("sample.mp4", args.output)