num image

app.py

@@ -111,7 +111,7 @@ def delete_old_files():
 threading.Thread(target=delete_old_files, daemon=True).start()
 
 
-@spaces.GPU(duration=120)  # [uncomment to use ZeroGPU]
+@spaces.GPU(duration=180)  # [uncomment to use ZeroGPU]
 def infer(prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps,
           progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
@@ -173,6 +173,24 @@ with gr.Blocks() as demo:
             run_button = gr.Button("Run", scale=1)
         result = gr.Image(label="Result", show_label=False)
 
+        num_images = gr.Radio(
+            choices=[1, 2, 4],
+            label="Number of Images",
+            value=1
+        )
+        result = gr.Gallery(label="Results", show_label=True, columns=2, rows=2)
+
+        MAX_PIXELS = 2 ** 21
+
+        def update_max_height(width):
+            max_height = MAX_PIXELS // width
+            return gr.update(maximum=max_height)
+
+
+        def update_max_width(height):
+            max_width = MAX_PIXELS // height
+            return gr.update(maximum=max_width)
+
         with gr.Accordion("Advanced Settings", open=False):
             seed = gr.Slider(
                 label="Seed",
@@ -189,7 +207,7 @@ with gr.Blocks() as demo:
                     label="Width",
                     minimum=512,
                     maximum=2048,
-                    step=16,
+                    step=32,
                     value=1024,
                 )
 
@@ -197,10 +215,11 @@ with gr.Blocks() as demo:
                     label="Height",
                     minimum=512,
                     maximum=2048,
-                    step=16,
+                    step=32,
                     value=1024,
                 )
-
+            width.change(update_max_height, inputs=[width], outputs=[height])
+            height.change(update_max_width, inputs=[height], outputs=[width])
             with gr.Row():
                 guidance_scale = gr.Slider(
                     label="Guidance scale",
@@ -217,9 +236,8 @@ with gr.Blocks() as demo:
                     step=1,
                     value=50,
                 )
-
         with gr.Column():
-            gr.Markdown("### Examples (Enhance prompt finish.)")
+            gr.Markdown("### Examples (Enhance prompt finish)")
             for i, ex in enumerate(examples):
                 with gr.Row():
                     ex_btn = gr.Button(
@@ -229,7 +247,7 @@ with gr.Blocks() as demo:
                         scale=3
                     )
                     ex_img = gr.Image(
-                        value=f"img/img_{i + 1}.png",
+                        value=f"../img/img_{i + 1}.png",
                         label="Effect",
                         interactive=False,
                         height=130,
@@ -237,10 +255,23 @@ with gr.Blocks() as demo:
                         scale=1
                     )
                     ex_btn.click(fn=lambda ex=ex: ex, inputs=[], outputs=prompt)
+
+    def update_gallery_layout(num_images):
+        if num_images == 1:
+            return gr.update(columns=1, rows=1)
+        elif num_images == 2:
+            return gr.update(columns=2, rows=1)
+        elif num_images == 4:
+            return gr.update(columns=2, rows=2)
+        return gr.update(columns=2, rows=2)
+
+
+    num_images.change(update_gallery_layout, inputs=[num_images], outputs=[result])
+
     gr.on(
         triggers=[run_button.click, prompt.submit],
         fn=infer,
-        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs=[prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps, num_images],
         outputs=[result, seed]
     )