Update app.py

app.py

@@ -14,7 +14,7 @@ pipe = pipe.to(device)
 def genie (prompt, scale, steps, seed):
      generator = torch.Generator(device=device).manual_seed(seed)
      #images = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
-     low_res_latents = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
+     low_res_latents = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images[0]
      upscaled_image = upscaler(prompt=prompt, image=low_res_latents, num_inference_steps=20, guidance_scale=0, generator=generator).images[0]
      return (low_res_latents, upscaled_image)