Update app.py

app.py

@@ -159,8 +159,8 @@ def infer(
         initial_latents = pipe.prepare_latents(
             batch_size=1,
             num_channels_latents=pipe.transformer.in_channels,
-            height=pipe.transformer.sample_size[0],
-            width=pipe.transformer.sample_size[1],
+            height=pipe.transformer.sample_size,
+            width=pipe.transformer.sample_size,
             dtype=pipe.transformer.dtype,
             device=pipe.device,
             generator=generator,