Update app.py

app.py

@@ -302,6 +302,7 @@ def generate_video(prompt: str, seed: int, image: str = None) -> tuple[str, dict
     # CRITICAL CHANGE: Transpose *after* converting to numpy and taking output[0]
     #output = output.transpose(1, 2, 0, 3)  # (T, H, W, C)
     print(output.shape)
+    print(output[0].shape)
     save_dir = f"./result"
     os.makedirs(save_dir, exist_ok=True)
     video_out_file = f"{save_dir}/{seed}.mp4"