Update app.py

app.py

@@ -97,7 +97,8 @@ def generate_image(
     scale=5,
     steps=30,
     strength=0.7,
-    seed=-1):
+    seed=-1,
+    nums=1):
 
     if seed == -1:
         seed = random.randint(0, MAX_SEED)
@@ -124,6 +125,7 @@ def generate_image(
             num_inference_steps=steps,
             strength=strength,
             generator = generator,
+            num_images_per_prompt = nums,
         )
     else:
         image = pipe(
@@ -134,8 +136,9 @@ def generate_image(
             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
+            num_images_per_prompt = nums,
         )        
-    return image.images[0]
+    return image, seed
 
 
 examples = [
@@ -160,9 +163,8 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     gr.HTML("<p><center><a href='https://huggingface.co/stabilityai/stable-diffusion-3-medium'>sd3m</a> text/image-to-image generation<br>Update: img2img, add Strength, T5 long Token</center></p>")
     with gr.Group():
         with gr.Row():
-            prompt = gr.MultimodalTextbox(label='Enter Your Prompt (Multi-Languages)', interactive=True, placeholder="Enter prompt, add one image.", file_types=['image'], scale=6)
-            submit = gr.Button(scale=1, variant='primary')
-    img = gr.Image(label='SD3M Generated Image')
+            prompt = gr.MultimodalTextbox(label='Enter Your Prompt (Multi-Languages)', interactive=True, placeholder="Enter prompt, add one image.", file_types=['image'], scale=1)
+    img = gr.Gallery(label='SD3M Generated Image',columns = 1, preview=True)
     with gr.Accordion("Advanced Options", open=False):
         with gr.Row():
             negative = gr.Textbox(label="Negative prompt", value="low quality, ugly, blurry, poor face, bad anatomy")
@@ -210,7 +212,14 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
                 maximum=10000000000000,
                 step=1,
                 value=-1,
-            )      
+            )
+            nums = gr.Slider(
+                label="Image Numbers",
+                minimum=1,
+                maximum=4,
+                step=1,
+                value=1,
+            )  
     gr.Examples(
         examples=examples,
         inputs=prompt,
@@ -221,12 +230,9 @@ with gr.Blocks(css=CSS, js=JS, theme="soft") as demo:
     )
 
     prompt.submit(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps, strength, seed],
-                 outputs=img,
-                 )
-    submit.click(fn=generate_image,
-                 inputs=[prompt, negative, width, height, scale, steps, strength, seed],
-                 outputs=img,
+                 inputs=[prompt, negative, width, height, scale, steps, strength, seed, nums],
+                 outputs=[img,seed],
                  )
+
     
 demo.queue().launch()