Update app.py

app.py

@@ -50,13 +50,14 @@ class calculateDuration:
         else:
             print(f"Elapsed time: {self.elapsed_time:.6f} seconds")
 
-def update_selection(evt: gr.SelectData, width, height):
+def update_selection(evt: gr.SelectData, width, height, default_scale, lora_scale):
     selected_lora = loras[evt.index]
     new_placeholder = f"Type a prompt for {selected_lora['title']}"
     prompt = selected_lora["prompt"]
     lora_repo = selected_lora["repo"]
     updated_text = f"### Selected: [{lora_repo}](https://huggingface.co/{lora_repo}) ✨"
-    lora_scale = selected_lora["lora_scale"]
+    if default_scale:
+        lora_scale = selected_lora["lora_scale"]
     if "aspect" in selected_lora:
         if selected_lora["aspect"] == "portrait":
             width = 768
@@ -197,11 +198,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                 with gr.Row():
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
-                    #lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95)
+
+                    default_scale = gr.Checkbox(True, label="Use default LoRA scale")
+                    lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=3, step=0.01, value=0.95)
 
     gallery.select(
         update_selection,
-        inputs=[width, height],
+        inputs=[width, height, default_scale, lora_scale],
         outputs=[prompt, selected_info, selected_index, width, height, lora_scale]
     )
     gr.on(