Update app.py

app.py

@@ -108,7 +108,7 @@ else:
 print("Initialization complete. Gradio is starting...")
 
 @spaces.GPU(duration = 45)
-def generate(prompt, negative_prompt, width, height, num_inference_steps, optional_lora_id, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, negative_prompt, width, height, num_inference_steps, optional_lora_id, lora_weights, progress=gr.Progress(track_tqdm=True)):
     
     # --- Step 1: ALWAYS build the full weight vector from scratch for THIS run ---
     
@@ -126,7 +126,7 @@ def generate(prompt, negative_prompt, width, height, num_inference_steps, option
         if internal_name_to_add:
             try:
                 optional_lora_index = ALL_ADAPTER_NAMES.index(internal_name_to_add)
-                adapter_weights[optional_lora_index] = 1.0
+                adapter_weights[optional_lora_index] = lora_weights
             except ValueError:
                 print(f"Warning: Could not find index for selected LoRA '{internal_name_to_add}'. It will not be applied.")
 
@@ -165,7 +165,8 @@ iface = gr.Interface(
         gr.Slider(minimum=1, maximum=80, step=1, label="Inference Steps", value=10),
         gr.Textbox(
             label="Optional LoRA",
-        )
+        ),
+        gr.Slider(label="LoRA Weights", minimum=0, maximum=2, step=0.01, value=1)
     ],
     outputs=gr.Image(label="output"),
 )