Update app.py

app.py

@@ -36,7 +36,10 @@ def run_lora(prompt, cfg_scale, steps, selected_index, seed, width, height, lora
     trigger_word = selected_lora["trigger_word"]
 
     # Load LoRA weights
-    pipe.load_lora_weights(lora_path)
+    if "weights" in selected_lora:
+        pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"])
+    else:
+        pipe.load_lora_weights(lora_path)
 
     # Set random seed for reproducibility
     generator = torch.Generator(device="cuda").manual_seed(seed)