Update app.py

app.py

@@ -141,21 +141,23 @@ def generate_image(
 with gr.Blocks() as demo:
     gr.Markdown("# Wan 2.2 (14B) Image")
     gr.Markdown("generate high quality images with Wan 2.2 14B")
-    with gr.Row():
-        with gr.Column():
-            prompt_input = gr.Textbox(label="Prompt", value=default_prompt_t2v)
-            with gr.Accordion("Advanced Settings", open=False):
-                negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, lines=3)
-                seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
-                randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
-                steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=27, label="Inference Steps") 
-                guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=3.5, label="Guidance Scale - high noise stage")
-                guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=4, label="Guidance Scale 2 - low noise stage")
-
-            generate_button = gr.Button("Generate Image", variant="primary")
-        with gr.Column():
-            img_output = gr.Image(label="Generated Image", interactive=False)
-    
+    with gr.Column():
+       with gr.Row():
+          prompt_input = gr.Textbox(show_label=False,
+                max_lines=1,
+                placeholder="Enter your prompt",
+                container=False,)
+          generate_button = gr.Button("Run", variant="primary", scale=0)
+       img_output = gr.Image(label="Generated Image", interactive=False)
+       with gr.Accordion("Advanced Settings", open=False):
+            negative_prompt_input = gr.Textbox(label="Negative Prompt", value=default_negative_prompt, lines=3)
+            seed_input = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, interactive=True)
+            randomize_seed_checkbox = gr.Checkbox(label="Randomize seed", value=True, interactive=True)
+            steps_slider = gr.Slider(minimum=1, maximum=30, step=1, value=27, label="Inference Steps") 
+            guidance_scale_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=3.5, label="Guidance Scale - high noise stage")
+            guidance_scale_2_input = gr.Slider(minimum=0.0, maximum=10.0, step=0.5, value=4, label="Guidance Scale 2 - low noise stage")
+
+            
     ui_inputs = [ 
         prompt_input,
         negative_prompt_input,