diff --git a/app.py b/app.py
index f976714..3e41c77 100644
--- a/app.py
+++ b/app.py
@@ -475,6 +475,7 @@ def get_edit_duration(
     width=None,
     rewrite_prompt=True,
     zerogpu_budget=0,
+    num_images_per_prompt=1,
     progress=None,
 ):
     if zerogpu_budget and int(zerogpu_budget) > 0: