fix: progress update on step

app.py

@@ -44,14 +44,14 @@ def generate(prompt: str, progress=gr.Progress()):
         num_inference_steps=n_steps,
         denoising_end=high_noise_frac,
         output_type="latent",
-        callback=lambda s, t, d: progress((s, 100), desc="Generating first frame..."),
+        callback_on_step_end=lambda p, s, t, d: progress((s, 100), desc="Generating first frame..."),
     ).images[0]
     image = refiner(
         prompt=prompt,
         num_inference_steps=n_steps,
         denoising_start=high_noise_frac,
         image=image,
-        callback=lambda s, t, d: progress((s+40, 100), desc="Refining first frame..."),
+         callback_on_step_end=lambda p, s, t, d: progress((s+40, 100), desc="Refining first frame..."),
     ).images[0]
     image = to_tensor(image)
     frames: list[Image.Image] = pipeline(
@@ -62,7 +62,7 @@ def generate(prompt: str, progress=gr.Progress()):
         guidance_scale=9.0,
         generator=generator,
         decode_chunk_size=10,
-        callback=lambda s, t, d: progress((s+50, 100), desc="Generating video..."),
+        callback_on_step_end=lambda p, s, t, d: progress((s+50, 100), desc="Generating video..."),
     ).frames[0]
     frames = [to_tensor(frame.convert("RGB")).mul(255).byte().permute(1, 2, 0) for frame in frames]
     frames = torch.stack(frames)