Update app.py

app.py

@@ -270,7 +270,7 @@ def generate_video(prompt: str, seed: int, image: str = None) -> tuple[str, dict
     output = (output.numpy() * 255).astype(np.uint8)
     # Correct Transpose: (1, C, T, H, W) -> (1, T, H, W, C)
     output = output.transpose(0, 2, 3, 4, 1)
-    output = output[0]  # Remove batch dimension: (T, H, W, C)
+    #output = output[0]  # Remove batch dimension: (T, H, W, C)
     save_dir = f"./result"
     os.makedirs(save_dir, exist_ok=True)
     video_out_file = f"{save_dir}/{seed}.mp4"