Update app.py

app.py

@@ -18,8 +18,9 @@ upscaler.enable_xformers_memory_efficient_attention()
 def genie (Prompt, negative_prompt, height, width, scale, steps, Seed, upscale):
     generator = torch.Generator(device=device).manual_seed(Seed)
     if upscale == "Yes":
-        low_res_latents = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
-        image = upscaler(Prompt, negative_prompt=negative_prompt, image=low_res_latents, num_inference_steps=5, guidance_scale=0, generator=generator).images[0]
+        image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
+        upscaled = upscaler(Prompt, negative_prompt=negative_prompt, image=image, num_inference_steps=25, guidance_scale=0, generator=generator).images[0]
+        return (image, upscaled)
     else:
         image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale, generator=generator).images[0]
     return (image, image)