Update app.py

app.py

@@ -159,16 +159,15 @@ def infer_with_lora(input_image, prompt, selected_index, custom_lora, seed=42, r
             lora_path = load_lora_weights(lora_to_use["repo"], lora_to_use["weights"])
             if lora_path:
                 pipe.load_lora_weights(lora_path, adapter_name="selected_lora")
+                pipe.set_adapters(["selected_lora"], adapter_weights=[lora_scale])
+                print(f"loaded: {lora_path} with scale {lora_scale}")
                 current_lora = lora_to_use
-
             
         except Exception as e:
             print(f"Error loading LoRA: {e}")
             # Continue without LoRA
-    
-    # Set LoRA scale if LoRA is loaded
-    if current_lora and hasattr(pipe, 'set_adapters'):
-        pipe.set_adapters("selected_lora", adapter_weights=[lora_scale])
+    else:
+        print(f"using already loaded lora: {lora_to_use}")
     
     input_image = input_image.convert("RGB")
     # Add trigger word to prompt
@@ -237,10 +236,9 @@ with gr.Blocks(css="custom.css") as demo:
     with gr.Row(elem_id="main_app"):
         with gr.Column(scale=4, elem_id="box_column"):
             with gr.Group(elem_id="gallery_box"):
-                input_image = gr.Image(label="Upload image for editing", type="pil", height=250)
+                input_image = gr.Image(label="Upload a picture of yourself", type="pil", height=300)
                 
                 gallery = gr.Gallery(
-                    [(item["image"], item["title"]) for item in flux_loras_raw],
                     label="Pick a LoRA",
                     allow_preview=False,
                     columns=3,
@@ -278,7 +276,7 @@ with gr.Blocks(css="custom.css") as demo:
                     minimum=0,
                     maximum=2,
                     step=0.1,
-                    value=1.0,
+                    value=1.5,
                     info="Controls the strength of the LoRA effect"
                 )
                 seed = gr.Slider(