Update app.py

app.py

@@ -394,27 +394,24 @@ def generate_video(input_image, prompt, height, width,
     export_to_video(output_frames_list, video_path, fps=FIXED_FPS)
     return video_path, current_seed
 
-with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
+# --- Gradio UI ---
+css="""
+#title{text-align: center}
+#title h1{font-size: 3em; display:inline-flex; align-items:center}
+#title img{width: 100px; margin-right: 0.5em}
+#col-container { margin: 0 auto; max-width: 1000px; } /* Increased max-width for gallery */
+#gallery .grid-wrap{height: 20vh !important; max-height: 250px !important;} 
+.custom_lora_card { border: 1px solid #e0e0e0; border-radius: 8px; padding: 10px; margin-top: 10px; background-color: #f9f9f9; }
+.card_internal { display: flex; align-items: center; }
+.card_internal img { margin-right: 1em; border-radius: 4px; }
+.card_internal div h4 { margin-bottom: 0.2em; }
+.card_internal div small { font-size: 0.9em; color: #555; }
+#lora_list_link { font-size: 90%; background: var(--block-background-fill); padding: 0.5em 1em; border-radius: 8px; display:inline-block; margin-top:10px;}
+"""
+
+with gr.Blocks(css=css, theme=gr.themes.Ocean(font=[gr.themes.GoogleFont("Lexend Deca"), "sans-serif"])) as demo:
     with gr.Column(elem_classes=["main-container"]):
-        gr.Markdown("# ⚡ FusionX Enhanced Wan 2.1 I2V (14B)")
-
-        # Enhanced badges for FusionX
-        gr.HTML("""
-        <div class="badge-container">
-            <a href="https://huggingface.co/vrgamedevgirl84/Wan14BT2VFusioniX" target="_blank">
-                <img src="https://img.shields.io/static/v1?label=FusionX&message=ENHANCED%20MODEL&color=%236a4c93&labelColor=%23533a7d&logo=huggingface&logoColor=%23ffffff&style=for-the-badge" alt="FusionX Enhanced">
-            </a>
-            <a href="https://huggingface.co/spaces/Heartsync/WAN2-1-fast-T2V-FusioniX" target="_blank">
-                <img src="https://img.shields.io/static/v1?label=BASE&message=WAN%202.1%20T2V-FusioniX&color=%23008080&labelColor=%23533a7d&logo=huggingface&logoColor=%23ffffff&style=for-the-badge" alt="Base Model">
-            </a>
-            <a href="https://huggingface.co/spaces/Heartsync/WAN2-1-fast-T2V-FusioniX2" target="_blank">
-                <img src="https://img.shields.io/static/v1?label=BASE&message=WAN%202.1%20T2V-Fusioni2X&color=%23008080&labelColor=%23533a7d&logo=huggingface&logoColor=%23ffffff&style=for-the-badge" alt="Base Model">
-            </a>            
-            <a href="https://huggingface.co/spaces/Heartsync/wan2-1-fast-security" target="_blank">
-                <img src="https://img.shields.io/static/v1?label=WAN%202.1&message=FAST%20%26%20Furios&color=%23008080&labelColor=%230000ff&logo=huggingface&logoColor=%23ffa500&style=for-the-badge" alt="badge">
-            </a>
-        </div>
-        """)
+        gr.Markdown("# ✨ Fast 4 steps Wan 2.1 I2V (14B) with CausVid LoRA + Audio")
         
         
         with gr.Row():