Update app.py

app.py

@@ -349,7 +349,8 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
 
         ori_lengths = [video.size(1) for video in cond_latents]
         input_videos = cond_latents
-        cond_latents = [cut_videos(video, sp_size) for video in cond_latents]
+        if is_video:
+            cond_latents = [cut_videos(video, sp_size) for video in cond_latents]
 
         # runner.dit.to("cpu")
         print(f"Encoding videos: {list(map(lambda x: x.size(), cond_latents))}")
@@ -429,7 +430,7 @@ with gr.Blocks(title="SeedVR2: One-Step Video Restoration via Diffusion Adversar
         fps = gr.Number(label="fps", value=24)
 
     with gr.Row():
-        output_video = gr.Video(label="Output")
+        output_video = gr.Video(label="Output_Video")
         output_image = gr.Image(label="Output_Image")
         download_link = gr.File(label="Download the output")