Update app.py

app.py

@@ -125,7 +125,7 @@ def generate(segment, image, prompt, size, guidance_scale, num_inference_steps,
             dtype=torch.float32, device=device, generator=generator, latents=None,
         )
         image_latents = pipe.image_latents(
-                image, batch_size, size, size, device, torch.float32, num_channels_latents, num_latent_frames
+                image, 1, size, size, device, torch.float32, num_channels_latents, num_latent_frames
         )
         image_latents = image_latents.to(pipe.transformer.dtype)
         guidance = torch.tensor([guidance_scale] * latents.shape[0], dtype=transformer_dtype, device=device) * 1000.0