Update app.py

app.py

@@ -134,6 +134,9 @@ def run_lora(prompt, cfg_scale, steps, selected_index, seed, width, height, lora
 
     return image
 
+'''
+#gen_btn{height: 100%}
+'''
 with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown("# FLUX.1 LoRA the Explorer")
     selected_index = gr.State(None)
@@ -141,7 +144,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
         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):
-            generate_button = gr.Button("Generate", variant="primary")
+            generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
     with gr.Row():
         with gr.Column(scale=2):
             selected_info = gr.Markdown("")
@@ -174,8 +177,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
                 lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, step=0.01, value=1)
 
     gallery.select(update_selection, outputs=[prompt, selected_info, selected_index])
-    
-    generate_button.click(
+
+    gr.on(
+        triggers=[generate_button.click, prompt.submit],
         fn=run_lora,
         inputs=[prompt, cfg_scale, steps, selected_index, seed, width, height, lora_scale],
         outputs=[result]