Update app.py

app.py

@@ -45,7 +45,7 @@ def make_prompt(text: str) -> list[str]:
 def randomize_seed_fn(seed: int, randomize: bool) -> int:
     return random.randint(0, MAX_SEED) if randomize else seed
 
-def generate_image(prompt: str, seed: int, guidance_scale: float, randomize: bool) -> list[Image.Image]:
+def generate_image(prompt: str, seed: int, guidance_scale: float, randomize: bool, progress: gr.Progress = gr.Progress(track_tqdm=True)) -> list[Image.Image]:
     seed = randomize_seed_fn(seed, randomize)
     set_global_seed(seed)
     formatted = make_prompt(prompt)
@@ -55,7 +55,7 @@ def generate_image(prompt: str, seed: int, guidance_scale: float, randomize: boo
         images.append(out.image)
     return images
 
-def process_image(prompt: str, img: Image.Image) -> str:
+def process_image(prompt: str, img: Image.Image, progress: gr.Progress = gr.Progress(track_tqdm=True)) -> str:
     messages = [{
         "role": "user",
         "content": [