Update gradio_app.py

gradio_app.py

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