Update app.py

app.py

@@ -35,9 +35,19 @@ pipe = StableDiffusionXLControlNetPipeline.from_pretrained(
 )
 
 # Load lora (giving it a name makes it active when using the name in the prompt)
-#pipe.load_lora_weights("ostris/ikea-instructions-lora-sdxl", weight_name="ikea_instructions_xl_v1_5.safetensors", adapter_name="ikea")
+pipe.load_lora_weights("ostris/ikea-instructions-lora-sdxl", weight_name="ikea_instructions_xl_v1_5.safetensors", adapter_name="ikea")
 pipe.load_lora_weights("nerijs/pixel-art-xl", weight_name="pixel-art-xl.safetensors", adapter_name="pixel")
-pipe.set_adapters("pixel")
+pipe.disable_lora()
+
+def activate_ikea_lora():
+    pipe.set_adapters("ikea")
+
+def activate_pixel_lora():
+    pipe.set_adapters("pixel")
+
+def disable_loras():
+    pipe.disable_lora()
+
 pipe.to(device)
 
 # 📸 Edge detection function using OpenCV (Canny)
@@ -82,6 +92,11 @@ with gr.Blocks() as demo:
             strength = gr.Slider(0.1, 1.0, value=0.8, label="Denoising Strength")
             guidance = gr.Slider(1, 20, value=7.5, label="Guidance Scale (Creativity)")
             controlnet_conditioning_scale = gr.Slider(0, 1, value=0.5, step=0.01, label="ControlNet Conditioning Scale")
+
+            with gr.Row():
+                ikea_lora_button = gr.Button("IKEA Instructions")
+                pixel_lora_button = gr.Button("Pixel Art")
+                disable_lora_button = gr.Button("Disable LoRas")
             
             generate_button = gr.Button("Generate Styled Image")
 
@@ -89,12 +104,27 @@ with gr.Blocks() as demo:
             edge_output = gr.Image(label="Edge Detected Image")
             result_output = gr.Image(label="Generated Styled Image")
 
-    # 🔗 Button Action
+    # 🔗 Generate Button Action
     generate_button.click(
         fn=generate_image,
         inputs=[prompt, input_image, low_threshold, high_threshold, strength, guidance, controlnet_conditioning_scale],
         outputs=[edge_output, result_output]
     )
 
+    ikea_lora_button.click(
+        fn = activate_ikea_lora,
+    )
+
+    pixel_lora_button.click(
+        fn = activate_pixel_lora,
+    )
+
+    disable_lora_button.click(
+        fn = disable_loras,
+    )
+    
+
+    
+
 # 🚀 Launch the app
 demo.launch()