app.py

@@ -49,10 +49,11 @@ def run_model(prompt, cfg_scale, steps, randomize_seed, seed, width, height):
     return image, seed  
 
 with gr.Blocks() as app:
-    gr.Row():
+    with gr.Row():
         with gr.Column():
             prompt = gr.Textbox(label="Prompt", placeholder="Type a prompt here")
             generate_button = gr.Button("Generate")
+    
     with gr.Row():
         result = gr.Image(label="Generated Image")
 
@@ -70,4 +71,4 @@ with gr.Blocks() as app:
         inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height],
         outputs=[result, seed],
         live=True
-    ).launch()
+    ).launch()