testing with 2 loras

app.py

@@ -49,7 +49,7 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, schedule
 
     # Load LoRA weights
     pipe.load_lora_weights("Abdullah-Habib/logolora",scale = 1)
-    pipe.load_lora_weights("Abdullah-Habib/icon-lora",scale = 0.5)
+    # pipe.load_lora_weights("Abdullah-Habib/icon-lora",scale = 0.5)
 
     # Set scheduler
     scheduler_config = pipe.scheduler.config
@@ -105,8 +105,24 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, schedule
 
     # Unload LoRA weights
     pipe.unload_lora_weights()
+    pipe.load_lora_weights("Abdullah-Habib/icon-lora", scale=1)
 
-    return image
+    # Pass the generated logo through the icon LoRA
+    rounded_square_image = pipe(
+        prompt="rounded square icon",
+        image=logo_image,  # Pass the generated logo image
+        num_inference_steps=steps,
+        guidance_scale=cfg_scale,
+        width=width,
+        height=height,
+        generator=generator,
+    ).images[0]
+
+    # Unload the icon LoRA
+    pipe.unload_lora_weights()
+
+    return rounded_square_image
+    # return image
 
 with gr.Blocks(theme=gr.themes.Soft()) as app: