Update app.py

app.py

@@ -141,11 +141,9 @@ def generate(
     
     pipe.to(device)
 
-    upscaler_pipe = StableDiffusionXLImg2ImgPipeline(**pipe.components)
-
     try:
         gr.Info("Generating image...")
-        latents = pipe(
+        img = pipe(
             prompt = prompt,
             negative_prompt = negative_prompt,
             width = width,
@@ -154,20 +152,6 @@ def generate(
             num_inference_steps = num_inference_steps,
             generator = generator,
             num_images_per_prompt=1,
-            output_type="latent",
-        ).images
-
-        upscaled_latents = upscale(latents, "nearest-exact", 2.0)
-
-        gr.Info("Upscaling...")
-        img = upscaler_pipe(
-            prompt=prompt,
-            negative_prompt=negative_prompt,
-            image=upscaled_latents,
-            guidance_scale=guidance_scale,
-            num_inference_steps=num_inference_steps,
-            strength=0.55,
-            generator=generator,
             output_type="pil",
         ).images[0]
 
@@ -176,7 +160,6 @@ def generate(
     except Exception as e:
         print(f"An error occurred: {e}")
     finally:
-        del upscaler_pipe
         free_memory()
 
 if torch.cuda.is_available():