Upload app.py with huggingface_hub

app.py

@@ -8,9 +8,10 @@ import torch
 import gradio as gr
 import tempfile
 import time
+import numpy as np
+import imageio
 from omegaconf import OmegaConf
 from einops import rearrange
-from torchvision.io import write_video
 
 from pipeline import CausalInferencePipeline
 from utils.wan_wrapper import WanDiffusionWrapper, WanTextEncoder, WanVAEWrapper
@@ -219,10 +220,10 @@ def generate(
     video = rearrange(video, 'b t c h w -> b t h w c').cpu()
 
     # Save as MP4
+    video_np = (video[0].numpy() * 255).clip(0, 255).astype(np.uint8)
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as f:
         output_path = f.name
-
-    write_video(output_path, video[0], fps=16)
+    imageio.mimsave(output_path, video_np, fps=16, quality=8)
 
     elapsed = time.time() - start_time
     print(f"Generation completed in {elapsed:.2f}s")