Update app.py

app.py

@@ -370,20 +370,35 @@ def generate_video(
 
 # Create Gradio interface
 with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
-    gr.Markdown("# LTX-2 Distilled 🎥🔈: The First Open Source Audio-Video Model")
-    gr.Markdown("Fast, state-of-the-art video & audio generation with [Lightricks LTX-2 TI2V model](https://huggingface.co/Lightricks/LTX-2) and [distillation LoRA](https://huggingface.co/Lightricks/LTX-2/blob/main/ltx-2-19b-distilled-lora-384.safetensors) for accelerated inference. Read more: [[model]](https://huggingface.co/Lightricks/LTX-2), [[code]](https://github.com/Lightricks/LTX-2)")
+    gr.HTML(
+        """
+        <div style="text-align: left;">
+            <p style="font-size:16px; display: inline; margin: 0;">
+                <strong>LTX-2 Distilled</strong> 
+            </p>
+            <a href="https://huggingface.co/Lightricks/LTX-2" style="display: inline-block; vertical-align: middle; margin-left: 0.5em;">
+                [model]
+            </a>
+        </div>
+        <div style="text-align: left;">
+            <p style="font-size:16px; display: inline; margin: 0;">
+                Using FA3 and Gemma 3 12B 4bit Quantisation for Faster Inference
+            </p>
+        </div>
+        """
+    )
     with gr.Row():
         with gr.Column():
             input_image = gr.Image(
                 label="Input Image (Optional)",
-                type="pil"
-            )
+                type="pil",
+                height=512)
 
             prompt = gr.Textbox(
                 label="Prompt",
-                info="for best results - make it as elaborate as possible",
                 value="Make this image come alive with cinematic motion, smooth animation",
                 lines=3,
+                max_lines=3,
                 placeholder="Describe the motion and animation you want..."
             )
             with gr.Row():
@@ -396,11 +411,10 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                 )
                 enhance_prompt = gr.Checkbox(
                         label="Enhance Prompt",
-                        value=True
+                        value=True,
+                        visible=False
                     )
 
-            generate_btn = gr.Button("Generate Video", variant="primary", size="lg")
-
             with gr.Accordion("Advanced Settings", open=False):
                 seed = gr.Slider(
                     label="Seed",
@@ -428,7 +442,8 @@ with gr.Blocks(title="LTX-2 Video Distilled 🎥🔈") as demo:
                     )
 
         with gr.Column():
-            output_video = gr.Video(label="Generated Video", autoplay=True)
+            output_video = gr.Video(label="Generated Video", autoplay=True, height=512)
+            generate_btn = gr.Button("Generate Video", variant="primary", size="lg")
 
     generate_btn.click(
         fn=generate_video,