update

app.py

@@ -385,10 +385,14 @@ def generation_loop(video_path='./test_videos', seed=666, batch_size=1, cfg_scal
             sample = sample.clip(-1, 1).mul_(0.5).add_(0.5).mul_(255).round()
             sample = sample.to(torch.uint8).numpy()
 
+            mediapy.write_video(
+                output_dir, sample, fps=self.config.dit_generation.get("fps", 12)
+            )
+
+        print(f"Generated video size: {sample.size()}")
         gc.collect()
         torch.cuda.empty_cache()
-        print(f"Generated video size: {sample.size()}")
-        return sample, output_dir
+        return output_dir, output_dir
 
 
 title = "SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training"