Update gradio_app.py

gradio_app.py

@@ -191,6 +191,13 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
                 cn_videos.append(cn_video)
             
             cn_videos = torch.cat(cn_videos, dim=2)
+            if cn_videos.shape[2] > frames:
+                idxs = []
+                for i in range(y):
+                    index = int((i + 0.5) * cn.shape[2] / frames)
+                    idxs.append(min(index, x - 1))
+                cn_videos = cn_videos[:, :, idxs, :, :]
+                print("cn_videos.shape after slicing", cn_videos.shape)
             model_list = []
             for model in model_list:
                 model.control_scale = control_scale