Update generate.py

generate.py

@@ -1,6 +1,7 @@
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 import argparse
+import shutil
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--prompt", required=True)
@@ -9,8 +10,6 @@ args = parser.parse_args()
 
 pipe = pipeline(Tasks.text_to_video_synthesis, model='damo-vilab/modelscope-text-to-video-synthesis')
 result = pipe(args.prompt)
-video_path = result['output_video']
 
-# Rename or move to match expected filename
-import shutil
-shutil.move(video_path, args.output)
+# Move video file to expected output name
+shutil.move(result["output_video"], args.output)