Update app.py

app.py

@@ -352,8 +352,8 @@ with gr.Blocks() as demo:
     gr.Markdown("# Fast 4 steps Wan 2.2 I2V (14B) with Lightning LoRA")
     gr.Markdown("run Wan 2.2 in just 4-8 steps, with [Lightning LoRA](https://huggingface.co/Kijai/WanVideo_comfy/tree/main/Wan22-Lightning), fp8 quantization & AoT compilation - compatible with 🧨 diffusers and ZeroGPU⚡️")
 
-    with gr.Row(equal_height=True):   # ensures columns align in height
-        with gr.Column(scale=1, elem_classes=["stretch-col"]):
+    with gr.Row():   # ensures columns align in height
+        with gr.Column():
             input_image_component = gr.Image(
                 type="pil",
                 label="Input Image (auto-resized to target H/W)",
@@ -387,7 +387,7 @@ with gr.Blocks() as demo:
 
             generate_button = gr.Button("Generate Video", variant="primary")
 
-        with gr.Column(scale=1, elem_classes=["stretch-col"]):
+        with gr.Column():
             video_output = gr.Video(label="Generated Video", autoplay=True, interactive=False, elem_classes=["stretch-video"])
 
     ui_inputs = [