Update app.py

app.py

@@ -15,7 +15,29 @@ from torchao.quantization import quantize_
 from torchao.quantization import Float8DynamicActivationFloat8WeightConfig, Int8WeightOnlyConfig
 import aoti
 import uuid
-
+import imageio.v3 as iio
+
+
+def export_browser_safe_video(frames, path, fps=16):
+    """
+    frames: list of PIL images or numpy arrays (H, W, 3), uint8
+    path: output .mp4 path
+    """
+    # convert PIL to np if needed
+    np_frames = []
+    for f in frames:
+        if hasattr(f, "convert"):
+            f = f.convert("RGB")
+            f = np.array(f)
+        np_frames.append(f)
+
+    iio.imwrite(
+        path,
+        np_frames,
+        fps=fps,
+        codec="libx264",
+        pixelformat="yuv420p",  # important for browser support
+    )
 # =========================================================
 # MODEL CONFIGURATION
 # =========================================================
@@ -207,7 +229,7 @@ def generate_video(
 
     with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as tmpfile:
         video_path = tmpfile.name
-    export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
+    export_browser_safe_video(output_frames_list, video_path)
     hf_upload(video_path,prompt, repo="obsxrver/hf-space-output")
     return video_path, current_seed