Update app.py

app.py

@@ -36,7 +36,6 @@ logger = logging.getLogger(__name__)
 
 _predictor = None
 task_type = TaskType.I2V  # Default task type.
-
 @spaces.GPU(duration=90)
 def init_predictor():
     global _predictor
@@ -93,7 +92,7 @@ def generate_video(prompt, seed, image=None):
         assert image is not None, "Please input an image for I2V task."
         kwargs["image"] = Image.open(image)
     elif task_type == TaskType.T2V:
-        pass # No image
+        pass
     else:
       raise ValueError(f"Invalid task_type: {task_type}")
 
@@ -104,13 +103,13 @@ def generate_video(prompt, seed, image=None):
 
     # --- Convert to NumPy, move to CPU, scale, and change dtype ---
     output = (output.cpu().numpy() * 255).astype(np.uint8)
-    # --- Convert from B, C, T, H, W to B, T, H, W, C
-    output = output.transpose(0, 2, 3, 4, 1)
+    output = output.transpose(0, 2, 3, 4, 1) #Correct transpose.
+
     save_dir = f"./result/{task_type.name}"
     os.makedirs(save_dir, exist_ok=True)
     video_out_file = f"{save_dir}/{prompt[:100].replace('/','')}_{seed}.mp4"
     print(f"generate video, local path: {video_out_file}")
-    export_to_video(output, video_out_file, fps=24) # Pass fps
+    export_to_video(output, video_out_file, fps=24)
     return video_out_file, kwargs
 
 
@@ -143,7 +142,6 @@ if __name__ == "__main__":
         task_type = TaskType.T2V
     elif args.task_type == "i2v":
         task_type = TaskType.I2V
-    # No else, default is already set.
 
     demo = create_gradio_interface()
     demo.queue().launch()