Update app.py

app.py

@@ -43,8 +43,8 @@ import torch
 torch._dynamo.config.suppress_errors = True
 torch._dynamo.config.disable = True
 
-import spaces
 import gradio as gr
+import spaces
 import numpy as np
 from huggingface_hub import hf_hub_download, snapshot_download
 
@@ -266,13 +266,21 @@ RESOLUTIONS = {
 def get_duration(
     prompt: str,
     negative_prompt: str,
+    input_image,
     duration: float,
+    seed: int,
+    randomize_seed: bool,
     height: int,
     width: int,
-    num_frames: int,
     enhance_prompt: bool,
-    video_cfg: float,
-    audio_cfg: float,
+    video_cfg_scale: float,
+    video_stg_scale: float,
+    video_rescale_scale: float,
+    video_a2v_scale: float,
+    audio_cfg_scale: float,
+    audio_stg_scale: float,
+    audio_rescale_scale: float,
+    audio_v2a_scale: float,
     progress,
 ) -> int:
     """
@@ -497,7 +505,7 @@ css = """
 .progress-text {color: white}
 """
 
-with gr.Blocks(title="LTX-2.3 Two-Stage HQ Video Generation", css=css) as demo:
+with gr.Blocks(title="LTX-2.3 Two-Stage HQ Video Generation") as demo:
     gr.Markdown("# LTX-2.3 Two-Stage HQ Video Generation")
     gr.Markdown(
         "High-quality text/image-to-video generation using the dev model + distilled LoRA. "
@@ -720,4 +728,9 @@ with gr.Blocks(title="LTX-2.3 Two-Stage HQ Video Generation", css=css) as demo:
 # =============================================================================
 
 if __name__ == "__main__":
-    demo.queue().launch(mcp_server=True)
+    demo.queue().launch(
+        theme=gr.themes.Citrus(),
+        css=css,
+        mcp_server=True,
+        share=True,
+    )