Update app.py

app.py

@@ -14,9 +14,9 @@ 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[0]
+     low_res_latents = pipe(prompt, num_inference_steps=steps, guidance_scale=scale, generator=generator, output_type="latent").images
      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)
+     return upscaled_image
     
 gr.Interface(fn=genie, inputs=[gr.Textbox(label='What you want the AI to generate. 77 Token Limit.'), 
                                gr.Slider(1, maximum=15, value=10, step=.25),