Update app.py

app.py

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