Update app.py

app.py

@@ -39,7 +39,7 @@ css = """
 }
 """
 
-def generate(prompt, neg_prompt, samp_steps, guide_scale, batch_size, seed, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, neg_prompt, samp_steps, guide_scale, batch_size, seed, height, width, progress=gr.Progress(track_tqdm=True)):
     if seed < 0:
         seed = random.randint(1,999999)
     images = pipe(
@@ -49,7 +49,8 @@ def generate(prompt, neg_prompt, samp_steps, guide_scale, batch_size, seed, prog
         guidance_scale=guide_scale,
         #cross_attention_kwargs={"scale": lora_scale},
         num_images_per_prompt=lora_scale,
-        width=600,
+        height=height,
+        width=width,
         generator=torch.manual_seed(seed),
     ).images
     return [(img, f"Image {i+1}") for i, img in enumerate(images)]
@@ -64,10 +65,13 @@ with gr.Blocks(css=css) as demo:
             samp_steps = gr.Slider(1, 50, value=20, step=1, label="Sampling steps")
             guide_scale = gr.Slider(1, 6, value=3, step=0.5, label="Guidance scale")
             batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size")
-        seed = gr.Number(label="seed", value="-1", precision=0)
+        with gr.Row():
+            seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=999999, step=1)
+            height = gr.Slider(label="Height", value=1024, minumum-512, maximum=2048, step=8)
+            width = gr.Slider(label="Width", value=1024, minumum-512, maximum=2048, step=8)
         gallery = gr.Gallery(label="Generated images", height=800)
 
-    submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed], [gallery], queue=True)
+    submit_btn.click(generate, [prompt, negative_prompt, samp_steps, guide_scale, batch_size, seed, height, width], [gallery], queue=True)
 
 demo.queue(1)
 demo.launch(debug=True)