Update app.py

app.py

@@ -124,7 +124,7 @@ def generate_video(
     cfg_guidance_scale: float,
     height: int,
     width: int,
-    progress=gr.Progress()
+    progress=gr.Progress(track_tqdm=True)
 ):
     """Generate a video based on the given parameters."""
     try:
@@ -155,8 +155,7 @@ def generate_video(
             # Format: (image_path, frame_idx, strength)
             images = [(str(temp_image_path), 0, 1.0)]
 
-        # Run inference
-        progress(0, desc="Generating video (2-stage)...")
+        # Run inference - progress automatically tracks tqdm from pipeline
         pipeline(
             prompt=prompt,
             negative_prompt=negative_prompt,
@@ -172,7 +171,6 @@ def generate_video(
             tiling_config=TilingConfig.default(),
         )
 
-        progress(1.0, desc="Done!")
         return str(output_path)
 
     except Exception as e:
@@ -294,4 +292,4 @@ with gr.Blocks(title="LTX-2 Image-to-Video") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()