Update app.py

app.py

@@ -58,7 +58,7 @@ def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height
         width=width,
         generator=torch.manual_seed(seed),
     ).images
-    return [(img, f"Image {i+1}") for i, img in enumerate(images)]
+    return [(img, f"Image {i+1}") for i, img in enumerate(images), seed]
         
 def set_base_model(base_model_id):
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
@@ -106,16 +106,19 @@ with gr.Blocks(css=css) as demo:
         with gr.Row():
             samp_steps = gr.Slider(1, 50, value=20, step=1, label="Sampling steps")
             cfg_scale = gr.Slider(1, 10, value=3, step=0.5, label="Guidance scale")
-            batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size")
+            batch_size = gr.Slider(1, 4, value=1, step=1, label="Batch size")
         with gr.Row():
-            height = gr.Slider(label="Height", value=1024, minimum=8, maximum=2560, step=8)
+            height = gr.Slider(label="Height", value=1024, minimum=8, maximum=1536, step=8)
             width = gr.Slider(label="Width", value=1024, minimum=8, maximum=2560, step=8)
         with gr.Row():
             pixels = gr.Number(label="Pixel Ratio", value=1, interactive=False)
             seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
         gallery = gr.Gallery(label="Generated images")
+        with gr.Row():
+            lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
+            
         ex = gr.Examples(examples=examples, inputs=[prompt, negative_prompt])
-    submit_btn.click(generate, [prompt, negative_prompt, samp_steps, cfg_scale, batch_size, seed, height, width], [gallery], queue=True)
+    submit_btn.click(generate, [prompt, negative_prompt, samp_steps, cfg_scale, batch_size, seed, height, width], [gallery, lastSeed], queue=True)
     height.release(update_pixel_ratio, [height, width], [pixels, height], queue=False)
     width.release(update_pixel_ratio, [width, height], [pixels, width], queue=False)