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"
-    # 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 for {selected_lora['title']}"
+    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,8 +43,8 @@ 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"]
+    selected_lora = loras[selected_index]
+    # lora_path = selected_lora["repo"]
     # trigger_word = selected_lora["trigger_word"]
 
     # Load LoRA weights
@@ -155,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])
+    gallery.select(update_selection, outputs=[prompt, selected_info, selected_index])
     
     generate_button.click(
         fn=run_lora,