update app

app.py

@@ -79,7 +79,6 @@ print("Loading FLUX.2 Klein 9B model base...")
 pipe = Flux2KleinPipeline.from_pretrained(
     "black-forest-labs/FLUX.2-klein-9B",
     torch_dtype=torch.bfloat16,
-    attn_implementation="kernels-community/flash-attn3",
 ).to(device)
 print("Base Model loaded successfully.")
 
@@ -146,7 +145,6 @@ def infer(
     width, height = update_dimensions_on_upload(input_image)
     processed_input = input_image.resize((width, height), Image.LANCZOS).convert("RGB")
     
-    # 4. Inference
     try:
         image = pipe(
             image=processed_input, 
@@ -200,31 +198,32 @@ with gr.Blocks() as demo:
                 input_image = gr.Image(
                     label="Upload Image", 
                     type="pil", 
-                    height=290, 
+                    height=300, 
                     sources=["upload", "webcam", "clipboard"]
                 )
-                prompt = gr.Text(
-                    label="Guiding Prompt", 
-                    show_label=True, 
-                    placeholder="e.g., a man with a red superhero mask"
-                )
-                
-                lora_adapter = gr.Dropdown(
-                    label="Choose a Creative Style",
-                    choices=list(ADAPTER_SPECS.keys()),
-                    value="Klein-Delight-Style"
-                )
+                with gr.Row():
+                    prompt = gr.Text(
+                        label="Edit Prompt", 
+                        show_label=True, 
+                        placeholder="e.g., a man with a red superhero mask"
+                    )
+                with gr.Row():
+                    lora_adapter = gr.Dropdown(
+                        label="Choose Editing Style",
+                        choices=list(ADAPTER_SPECS.keys()),
+                        value="Klein-Delight-Style"
+                    )
 
                 run_button = gr.Button("Apply Style", variant="primary")
 
-                with gr.Accordion("Advanced Settings", open=False):
+                with gr.Accordion("Advanced Settings", open=False, visible=False):
                     seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0)
                     randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
                     guidance_scale = gr.Slider(label="Guidance Scale", minimum=0.0, maximum=10.0, step=0.1, value=1.0)       
                     steps = gr.Slider(label="Steps", minimum=1, maximum=50, value=4, step=1)
                     
             with gr.Column():
-                output_image = gr.Image(label="Stylized Image", interactive=False, format="png", height=450)
+                output_image = gr.Image(label="Output Image", interactive=False, format="png", height=420)
                 used_seed = gr.Textbox(label="Used Seed", interactive=False, visible=False)
 
         gr.Examples(