Update app.py

app.py

@@ -19,11 +19,11 @@ pipe = DiffusionPipeline.from_pretrained(
 pipe.to("cuda")
 pipe.load_lora_weights(prj_path, weight_name="pytorch_lora_weights.safetensors")
 
-#refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
-#    "stabilityai/stable-diffusion-xl-refiner-1.0",
-#    torch_dtype=torch.float16,
-#)
-#refiner.to("cuda")
+refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-refiner-1.0",
+   torch_dtype=torch.float16,
+)
+refiner.to("cuda")
 
 
 def generate_image(prompt, seed=None):
@@ -33,7 +33,7 @@ def generate_image(prompt, seed=None):
     seed = int(seed)
     generator = torch.Generator("cuda").manual_seed(seed)
     image = pipe(prompt=prompt, generator=generator, num_inference_steps=25).images[0]
-    #image = refiner(prompt=prompt, generator=generator, image=image).images[0]
+    image = refiner(prompt=prompt, generator=generator, image=image).images[0]
     name = f"{seed}_{str(uuid.uuid4())}.png"
     save_path = f"images/{name}"
     image.save(save_path)