Update app.py

app.py

@@ -15,18 +15,17 @@ css = """
 }
 """
 
-def generate(prompt, samp_steps, batch_size, seed, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
     if seed < 0:
         seed = random.randint(1,999999)
-    images = txt2img(
+    image = txt2img(
         prompt,
         num_inference_steps=1,
-        num_images_per_prompt=batch_size,
         guidance_scale=0.0,
         generator=torch.manual_seed(seed),
     ).images
-    upscaled_images = [img.resize((1024,1024), 1) for img in images]
-    final_images = img2img(
+    upscaled_image = image.resize((1024,1024), 1)
+    final_image = img2img(
         prompt,
         num_inference_steps=samp_steps,
         guidance_scale=5,
@@ -57,13 +56,12 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             sampling_steps = gr.Slider(1, 20, value=5, step=1, label="Sampling steps")
-            batch_size = gr.Slider(1, 6, value=1, step=1, label="Batch size")
             seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
             lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
             
         gallery = gr.Gallery(show_label=False, preview=True, container=False, height=650)
         
-    submit_btn.click(generate, [prompt, sampling_steps, batch_size, seed], [gallery, lastSeed], queue=True)
+    submit_btn.click(generate, [prompt, sampling_steps, seed], [gallery, lastSeed], queue=True)
     
 txt2img, img2img = set_base_models()
 demo.launch(debug=True)