Update app.py

app.py

@@ -51,6 +51,7 @@ def set_timesteps(scheduler, num_inference_steps):
     scheduler.set_timesteps(num_inference_steps)
     scheduler.timesteps = scheduler.timesteps.to(torch.float32) # minor fix to ensure MPS compatibility, fixed in diffusers PR 3925
 
+batch_size = 1
 
 # Prep latents
 latents = torch.randn(