Update gradio_app.py

gradio_app.py

@@ -155,7 +155,7 @@ def get_image(image1, prompt, image2, dim_steps=50, ddim_eta=1., fs=None, seed=1
     image_tensor2 = transform(image2).unsqueeze(1) # [c,1,h,w]
     frame_tensor1 = repeat(image_tensor1, 'c t h w -> c (repeat t) h w', repeat=8)
     frame_tensor2 = repeat(image_tensor2, 'c t h w -> c (repeat t) h w', repeat=8)
-    videos = torch.cat([frame_tensor1, frame_tensor2], dim=1)
+    videos = torch.cat([frame_tensor1, frame_tensor2], dim=1).unsqueeze(0)
     # frame_tensor = torch.cat([frame_tensor1, frame_tensor1], dim=1)
     # _, filename = os.path.split(file_list[idx*2])