Update app.py

app.py

@@ -13,7 +13,7 @@ loaded_diffusion_model = from_pretrained_keras("keras-dreambooth/dreambooth-pira
 dreambooth_model._diffusion_model = loaded_diffusion_model
 
 
-def generate_images(prompt: str, negative_prompt:str, num_imgs_to_gen: int, ugs: int):
+def generate_images(prompt: str, negative_prompt:str, num_imgs_to_gen: int, num_steps: int, ugs: int):
     generated_img = dreambooth_model.text_to_image(
         prompt, 
         negative_prompt=negative_prompt,