testing with 2 loras

app.py

@@ -28,14 +28,14 @@ pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.float16)
 pipe.to("cuda")
 
 def update_selection(evt: gr.SelectData):
-    selected_lora = loras[evt.index]
-    new_placeholder = f"Type a prompt for {selected_lora['title']}"
-    lora_repo = selected_lora["repo"]
-    updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
+    # selected_lora = loras[evt.index]
+    new_placeholder = f"Type a prompt"
+    # lora_repo = selected_lora["repo"]
+    # updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
     return (
         gr.update(placeholder=new_placeholder),
         updated_text,
-        evt.index
+        # evt.index
     )
 
 @spaces.GPU
@@ -43,12 +43,13 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, schedule
     if selected_index is None:
         raise gr.Error("You must select a LoRA before proceeding.")
 
-    selected_lora = loras[selected_index]
-    lora_path = selected_lora["repo"]
-    trigger_word = selected_lora["trigger_word"]
+    # selected_lora = loras[selected_index]
+    # # lora_path = selected_lora["repo"]
+    # trigger_word = selected_lora["trigger_word"]
 
     # Load LoRA weights
-    pipe.load_lora_weights(lora_path)
+    pipe.load_lora_weights("Abdullah-Habib/logolora")
+    pipe.load_lora_weights("Abdullah-Habib/icon-lora")
 
     # Set scheduler
     scheduler_config = pipe.scheduler.config
@@ -92,7 +93,7 @@ def run_lora(prompt, negative_prompt, cfg_scale, steps, selected_index, schedule
 
     # Generate image
     image = pipe(
-        prompt=f"{prompt} {trigger_word}",
+        prompt=f"{prompt} logo, logoredmaf, icons",
         negative_prompt=negative_prompt,
         num_inference_steps=steps,
         guidance_scale=cfg_scale,
@@ -154,7 +155,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                 value="DPM++ 2M SDE Karras"
             )
 
-    gallery.select(update_selection, outputs=[prompt, selected_info, selected_index])
+    gallery.select(update_selection, outputs=[prompt, selected_info])
     
     generate_button.click(
         fn=run_lora,