Update app.py

app.py

@@ -22,12 +22,11 @@ MODEL_DIR = os.path.abspath("ckpts")
 HF_MAIN_REPO = "tencent/HunyuanVideo-1.5"
 HF_GLYPH_REPO = "multimodalart/glyph-sdxl-v2-byt5-small"
 HF_LLM_REPO = "Qwen/Qwen2.5-VL-7B-Instruct"
-HF_VISION_REPO = "black-forest-labs/FLUX.1-Redux-dev" # User specified
+HF_VISION_REPO = "black-forest-labs/FLUX.1-Redux-dev"
 
 # Configuration
 TRANSFORMER_VERSION = "480p_i2v_distilled"
 DTYPE = torch.bfloat16
-# ZeroGPU: Set False so we control offloading manually (CPU -> GPU -> CPU)
 ENABLE_OFFLOADING = False 
 
 def setup_environment():
@@ -303,25 +302,28 @@ def generate(input_image, prompt, length, steps, shift, seed, guidance, do_rewri
     return output_path, actual_prompt
 
 # --- Part 6: UI ---
-
+css = '''#col-container { max-width: 900px; margin: 0 auto; }
+.dark .progress-text{color: white !important}'''
 def create_ui():
-    with gr.Blocks(title="HunyuanVideo 1.5 I2V") as demo:
-        gr.Markdown(f"### 🎬 HunyuanVideo 1.5 I2V ({TRANSFORMER_VERSION})")
+    with gr.Blocks(title="HunyuanVideo 1.5 I2V", css=css) as demo:
+        gr.Markdown(f"#🎬 HunyuanVideo 1.5 I2V 480p distilled demo")
+        gr.Markdown(f"This is a demo for HunyuanVideo 1.5 I2v {TRANSFORMER_VERSION}, released together with a collection of 10 other checkpoints (text-to-video, 720p, upscalers). Check out the [HunyuanVideo-1.5 model page](https://huggingface.co/tencent/HunyuanVideo-1.5) for more")
         
         with gr.Row():
             with gr.Column():
-                img = gr.Image(label="Reference", type="pil", height=250)
+                img = gr.Image(label="Reference", type="pil")
                 prompt = gr.Textbox(label="Prompt", placeholder="Describe motion...", lines=2)
-                rewrite_chk = gr.Checkbox(label="Enable Prompt Rewrite (Recommended)", value=True)
+                rewrite_chk = gr.Checkbox(label="Enable Prompt Rewrite (Strongly Recommended)", value=True)
                 
-                with gr.Row():
-                    steps = gr.Slider(2, 50, value=6, step=1, label="Steps")
-                    guidance = gr.Slider(1.0, 5.0, value=1.0, step=0.1, label="Guidance")
-                with gr.Row():
-                    shift = gr.Slider(1.0, 20.0, value=5.0, step=0.5, label="Shift")
-                    length = gr.Slider(1, 129, value=61, step=4, label="Length")
-                    seed = gr.Number(value=-1, label="Seed", precision=0)
-                btn = gr.Button("Generate", variant="primary")
+                with gr.Accordion("Advanced Options", open=False):
+                    with gr.Row():
+                        steps = gr.Slider(2, 50, value=6, step=1, label="Steps")
+                        guidance = gr.Slider(1.0, 5.0, value=1.0, step=0.1, label="Guidance")
+                    with gr.Row():
+                        shift = gr.Slider(1.0, 20.0, value=5.0, step=0.5, label="Shift")
+                        length = gr.Slider(1, 129, value=61, step=4, label="Length")
+                        seed = gr.Number(value=-1, label="Seed", precision=0, info="-1 is a random seed")
+                    btn = gr.Button("Generate", variant="primary")
             
             with gr.Column():
                 out = gr.Video(label="Result", autoplay=True)