Update app.py

app.py

@@ -228,10 +228,10 @@ def infer(
             width=width,
             height=height,
          #   latents=None,
-          #  output='latent',
+            output='latent',
             generator=generator,
             max_sequence_length=512
-        ).images[0]
+        ).images
         print('-- got image --')
         sd35_image_image = pipe.vae.decode(sd_image / 0.18215).sample
         sd35_image = sd35_image.cpu().permute(0, 2, 3, 1).float().detach().numpy()