Update app.py

app.py

@@ -7,6 +7,24 @@ from diffusers import DiffusionPipeline, LCMScheduler
 from PIL import Image, ImageEnhance
 import io
 
+pipe = DiffusionPipeline.from_pretrained(
+    "stabilityai/stable-diffusion-xl-base-1.0",
+    variant="fp16",
+    torch_dtype=torch.float32
+).to("cuda")
+
+# set scheduler
+pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
+
+# Load and fuse lcm lora
+pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl", adapter_name="lcm")
+pipe.load_lora_weights("Pclanglais/wiki-model", weight_name="pytorch_lora_weights.safetensors", adapter_name="mickey")
+
+# Combine LoRAs
+pipe.set_adapters(["lcm", "mickey"], adapter_weights=[1.0, 1.0])
+pipe.fuse_lora()
+
+
 @spaces.GPU
 def generate_image(prompt, num_inference_steps, guidance_scale):
     model_id = "stabilityai/stable-diffusion-xl-base-1.0"
@@ -60,39 +78,13 @@ def inpaint_image(prompt, init_image, mask_image, num_inference_steps, guidance_
 
     return image
 
-def generate_image_with_adapter(prompt, num_inference_steps, guidance_scale):
-    pipe = DiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0",
-    variant="fp16",
-    torch_dtype=torch.float32
-    ).to("cuda")
-
-    # set scheduler
-    pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
-
-    # Load and fuse lcm lora
-    pipe.load_lora_weights("latent-consistency/lcm-lora-sdxl", adapter_name="lcm")
-    pipe.load_lora_weights("Pclanglais/Mickey-1928", weight_name="pytorch_lora_weights.safetensors", adapter_name="mickey")
-
-    # Combine LoRAs
-    pipe.set_adapters(["lcm", "mickey"], adapter_weights=[1.0, 0.8])
-    pipe.fuse_lora()
+def generate_image_with_adapter(pipe, prompt, num_inference_steps, guidance_scale):
     generator = torch.manual_seed(0)
     # Generate the image
     image = pipe(prompt=prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale, generator=generator).images[0]
-    pipe.unfuse_lora()
     return image
 
 
-def modify_image(image, brightness, contrast):
-    # Function to modify brightness and contrast
-    image = Image.open(io.BytesIO(image))
-    enhancer = ImageEnhance.Brightness(image)
-    image = enhancer.enhance(brightness)
-    enhancer = ImageEnhance.Contrast(image)
-    image = enhancer.enhance(contrast)
-    return image
-
 with gr.Blocks(gr.themes.Soft()) as demo:
     with gr.Row():
         image_output = gr.Image(label="Generated Image")