Update app.py

app.py

@@ -33,17 +33,22 @@ pipe = pipe.to("cuda")
 pipe.enable_model_cpu_offload()
 
 @spaces.GPU
-def generate_image(prompt, height, width, num_inference_steps, guidance_scale):
-    seed = random.randint(0, 18446744073709551615)
+def generate_image(prompt, negative_prompt, height, width, num_inference_steps, guidance_scale, num_images_per_prompt, use_random_seed, seed):
+    if use_random_seed:
+        seed = random.randint(0, 2**32 - 1)
+    else:
+        seed = int(seed)  # Ensure seed is an integer
+    
     image = pipe(
         prompt=prompt,
+        negative_prompt=negative_prompt,
         height=height,
         width=width,
         num_inference_steps=num_inference_steps,
         guidance_scale=guidance_scale,
-        num_images_per_prompt=1,
+        num_images_per_prompt=num_images_per_prompt,
         generator=torch.Generator(pipe.device).manual_seed(seed)
-    ).images[0]
+    ).images
     return image, seed
 
 description = """
@@ -61,14 +66,18 @@ iface = gr.Interface(
     fn=generate_image,
     inputs=[
         gr.Textbox(label="Prompt"),
-        gr.Slider(512, 1344, 1024, step=64, label="Height"),
-        gr.Slider(512, 1344, 1024, step=64, label="Width"),
+        gr.Textbox(label="Negative Prompt"),
+        gr.Slider(512, 2048, 1024, step=64, label="Height"),
+        gr.Slider(512, 2048, 1024, step=64, label="Width"),
         gr.Slider(20, 50, 20, step=1, label="Number of Inference Steps"),
         gr.Slider(1, 20, 5, step=0.5, label="Guidance Scale"),
+        gr.Slider(1, 4, 1, step=1, label="Number of images per prompt"),
+        gr.Checkbox(label="Use Random Seed", value=True),
+        gr.Number(label="Seed", value=0, precision=0),
     ],
     outputs=[
-        gr.Image(label="Generated Image"),
-        gr.Number(label="Seed")
+        gr.Gallery(label="Result", elem_id="gallery", show_label=False),
+        gr.Number(label="Seed Used")
     ],
     title="Kolors",
     description=description,