Update app.py

app.py

@@ -150,16 +150,19 @@ def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator,sup
         generator = generator).images
       pipe=pipe.to("cpu")
       upscaler=upscaler.to("cuda")
-      result = upscaler(
+      latents = upscaler(
         prompt=prompt,
         negative_prompt = neg_prompt,
         image=low_res_latents,
         num_inference_steps=20,
         guidance_scale=0,
+        output_type="latent",
         generator=generator,
-      )
+      ).images
       pipe=pipe.to("cuda")
       upscaler=upscaler.to("cpu")
+      with torch.no_grad():
+        result = pipeline.decode_latents(latents)
     else:  
       result = pipe(
         prompt,