Fixed multiple videos save

inference.py

@@ -332,7 +332,9 @@ def main():
         )
 
     for i in range(images.shape[0]):
-        video_np = images.squeeze(0).permute(1, 2, 3, 0).cpu().float().numpy()
+        # Gathering from B, C, F, H, W to C, F, H, W and then permuting to F, H, W, C
+        video_np = images[i].permute(1, 2, 3, 0).cpu().float().numpy()
+        # Unnormalizing images to [0, 255] range
         video_np = (video_np * 255).astype(np.uint8)
         fps = args.frame_rate
         height, width = video_np.shape[1:3]