Update app.py

app.py

@@ -56,11 +56,11 @@ def get_seed(seed):
         return random.randint(0, MAX_SEED)
 
 @spaces.GPU(duration=30)
-def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, seed=None):
+def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=DEFAULT_HEIGHT, width=DEFAULT_WIDTH, steps=1, guidance=0, number=1, seed=None):
     
     seed = get_seed(seed)
 
-    print(input, negative_input, height, width, steps, guidance, seed)
+    print(input, negative_input, height, width, steps, guidance, number, seed)
 
     model.to(DEVICE)
     parameters  = {
@@ -70,7 +70,7 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
         "width": width,
         "num_inference_steps": steps,
         "guidance_scale": guidance,
-        "num_images_per_prompt": 1,
+        "num_images_per_prompt": number,
         "cross_attention_kwargs": {"scale": 0.65},
         "generator": torch.Generator().manual_seed(seed),
         "use_resolution_binning": True,
@@ -94,6 +94,7 @@ with gr.Blocks(css=css) as main:
         width = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_WIDTH, label="Width")
         steps = gr.Slider(minimum=0, maximum=100, step=1, value=8, label="Steps")
         guidance = gr.Slider(minimum=0, maximum=100, step=0.001, value=3, label = "Guidance")
+        number = gr.Slider(minimum=1, maximum=4, step=1, value=1, label="Number")
         seed = gr.Textbox(lines=1, value="", label="Seed (Blank for random)")
         submit = gr.Button("▶")
         maintain = gr.Button("☁️")
@@ -101,7 +102,7 @@ with gr.Blocks(css=css) as main:
     with gr.Column():
         images = gr.Gallery(columns=1, label="Image")
             
-    submit.click(generate, inputs=[input, negative_input, height, width, steps, guidance, seed], outputs=[images], queue=False)
+    submit.click(generate, inputs=[input, negative_input, height, width, steps, guidance, number, seed], outputs=[images], queue=False)
     maintain.click(cloud, inputs=[], outputs=[], queue=False)
 
 main.launch(show_api=True)