Update app.py

app.py

@@ -372,15 +372,16 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css, delete_cache=(60, 3600)) as app:
     with gr.Row():
         with gr.Column(scale=3):
             prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Type a prompt after selecting a LoRA")
-        with gr.Column(scale=1, elem_id="gen_column"):
-            generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
-            randomize_button = gr.Button("🎲", variant="secondary")
+        with gr.Row(scale=1):
+            generate_button = gr.Button("Generate", variant="primary")
     with gr.Row():
-        with gr.Column():
+        with gr.Column(scale=1):
+            randomize_button = gr.Button("🎲", variant="secondary", scale=1)
+        with gr.Column(scale=3):
             selected_info_1 = gr.Markdown("Select a LoRA 1")
             lora_scale_1 = gr.Slider(label="LoRA 1 Scale", minimum=0, maximum=3, step=0.01, value=0.95)
             remove_button_1 = gr.Button("Remove LoRA 1")
-        with gr.Column():
+        with gr.Column(scale=3):
             selected_info_2 = gr.Markdown("Select a LoRA 2")
             lora_scale_2 = gr.Slider(label="LoRA 2 Scale", minimum=0, maximum=3, step=0.01, value=0.95)
             remove_button_2 = gr.Button("Remove LoRA 2")