Update app.py

app.py

@@ -46,11 +46,11 @@ def randomize_seed_fn(seed: int, randomize: bool) -> int:
     return random.randint(0, MAX_SEED) if randomize else seed
 
 @spaces.GPU
-def generate_image(prompt: str, final_seed: int, guidance_scale: float, progress: gr.Progress = gr.Progress(track_tqdm=True)) -> list[Image.Image]:
+def generate_image(prompt: str, final_seed: int, guidance_scale: float, images_to_generate: int, progress: gr.Progress = gr.Progress(track_tqdm=True)) -> list[Image.Image]:
     set_global_seed(final_seed)
     formatted = make_prompt(prompt)
     images = []
-    for _ in range(4): # Original code generates 4 images
+    for _ in range(images_to_generate):
         out = pipe(formatted, guidance_scale=guidance_scale)
         images.append(out.image)
     return images
@@ -134,6 +134,11 @@ with gr.Blocks(title="BLIP3-o") as demo:
                         minimum=1.0, maximum=30.0,
                         step=0.5, value=3.0
                     )
+                    images_to_generate = gr.Slider(
+                        label="Number of images",
+                        minimum=1, maximum=4,
+                        step=1, value=2
+                    )
                     run_image_gen_btn = gr.Button("Generate Image")
 
                     text_gen_examples_data = [
@@ -215,7 +220,7 @@ with gr.Blocks(title="BLIP3-o") as demo:
 
     # Event listeners for Text -> Image
     # Chain seed randomization → run_generate_image_tab
-    gen_inputs = [prompt_gen_input, seed_slider, guidance_slider]
+    gen_inputs = [prompt_gen_input, seed_slider, guidance_slider, images_to_generate]
     
     run_image_gen_btn.click(
         fn=randomize_seed_fn,