fix: progress updater

app.py

@@ -67,6 +67,7 @@ def generate(prompt: str, progress=gr.Progress()):
         ),
     ).images[0]
     image = to_tensor(image)
+    progress((n_sdxl_steps + 1, total_steps), desc="Generating video...")
     frames: list[Image.Image] = pipeline(
         prompt=prompt,
         image=image,
@@ -74,17 +75,13 @@ def generate(prompt: str, progress=gr.Progress()):
         negative_prompt=negative_prompt,
         guidance_scale=9.0,
         generator=generator,
-        decode_chunk_size=10,
-        callback_on_step_end=create_progress_updater(
-            start=n_sdxl_steps,
-            total=total_steps,
-            desc="Generating video...",
-            progress=progress,
-        ),
+        decode_chunk_size=8,
+        num_frames=64,
     ).frames[0]
+    progress((total_steps - 1, total_steps), desc="Finalizing...")
     frames = [to_tensor(frame.convert("RGB")).mul(255).byte().permute(1, 2, 0) for frame in frames]
     frames = torch.stack(frames)
-    torchvision.io.write_video("video.mp4", frames, fps=8)
+    torchvision.io.write_video("video.mp4", frames, fps=16)
     return "video.mp4"
 
 app = gr.Interface(

utils.py

@@ -2,6 +2,6 @@ from gradio import Progress
 
 def create_progress_updater(start: int, total: int, desc: str, progress: Progress):
     def updater(pipe, step, timestep, callback_kwargs):
-        progress((step + start, total), desc=desc)
+        progress((step + start + 1, total), desc=desc)
         return callback_kwargs
     return updater