Update app.py

app.py

@@ -137,13 +137,13 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     # Load LoRA weights
     with calculateDuration(f"Loading LoRA weights for {selected_lora['title']}"):
         if "weights" in selected_lora:
-            pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name=selected_lora['title'], adapter_weights=lora_scale)
+            pipe.load_lora_weights(lora_path, weight_name=selected_lora["weights"], adapter_name=selected_lora["repo"], adapter_weights=lora_scale)
             pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast", adapter_weights=[1.0])
-            pipe.set_adapters(["fast", 'title'], adapter_weights=[1.0, lora_scale])
+            pipe.set_adapters({"fast", selected_lora["repo"]}, adapter_weights=[1.0, lora_scale])
         else:
-            pipe.load_lora_weights(lora_path, adapter_name=selected_lora['title'], adapter_weights=lora_scale)
+            pipe.load_lora_weights(lora_path, adapter_name=selected_lora["repo"], adapter_weights=lora_scale)
             pipe.load_lora_weights("ostris/OpenFLUX.1", weight_name="openflux1-v0.1.0-fast-lora.safetensors", adapter_name="fast", adapter_weights=[1.0])
-            pipe.set_adapters(["fast", 'title'], adapter_weights=[1.0, lora_scale])
+            pipe.set_adapters({"fast", selected_lora["repo"]}, adapter_weights=[1.0, lora_scale])
         
     # Set random seed for reproducibility
     with calculateDuration("Randomizing seed"):
@@ -210,13 +210,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
                     steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=5)
                 
                 with gr.Row():
-                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=768)
+                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
                     height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
                 
                 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=2.0, step=0.01, value=0.8)
+                    lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.0, step=0.01, value=0.85)
 
     gallery.select(
         update_selection,