Update app.py

app.py

@@ -71,6 +71,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
     return image
 
 '''
+#gen_column{height: 100%}
 #gen_btn{height: 100%}
 '''
 with gr.Blocks(theme=gr.themes.Soft()) as app:
@@ -79,7 +80,7 @@ with gr.Blocks(theme=gr.themes.Soft()) 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):
+        with gr.Column(scale=1, elem_id="gen_column"):
             generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
     with gr.Row():
         with gr.Column(scale=3):