Update generate.py

generate.py

@@ -22,6 +22,13 @@ command = f"python run_model.py --task {args.task} --size {args.size} --frame_nu
 
 subprocess.run(command, shell=True)
 
+process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+stdout, stderr = process.communicate()
+
+print("Output:", stdout.decode())
+print("Error:", stderr.decode())
+
+
 # Save output
 if os.path.exists("output.mp4"):
     print("✅ Video generated successfully: output.mp4")