Update app.py

app.py

@@ -386,9 +386,9 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
         gc.collect()
         torch.cuda.empty_cache()
         if is_image:
-            return output_dir, None
+            return output_dir, None, output_dir
         else:
-            return None, output_dir
+            return None, output_dir, output_dir
 
 
 with gr.Blocks(title="SeedVR2: One-Step Video Restoration via Diffusion Adversarial Post-Training") as demo: