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):
@@ -32,8 +32,8 @@ def generate_image(prompt, seed=None):
 
     seed = int(seed)
     generator = torch.Generator("cuda").manual_seed(seed)
-    image = pipe(prompt=prompt, generator=generator).images[0]
-    image = refiner(prompt=prompt, generator=generator, image=image).images[0]
+    image = pipe(prompt=prompt, generator=generator, num_inference_steps=25).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)