Update app.py

app.py

@@ -57,7 +57,7 @@ def generate(prompt, neg_prompt, samp_steps, cfg_scale, batch_size, seed, height
         width=width,
         generator=torch.manual_seed(seed),
     ).images
-    return gr.update(value = [(img, f"Image {i+1}") for i, img in enumerate(images)], height=height), seed
+    return gr.Gallery(value = [(img, f"Image {i+1}") for i, img in enumerate(images)], height=height), seed
         
 def set_base_model(base_model_id):
     vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
@@ -112,7 +112,7 @@ with gr.Blocks(css=css) as demo:
         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(show_label=False, elem_id="gallery", preview=True, container=False)
+        gallery = gr.Gallery(show_label=False, preview=True, container=False)
         with gr.Row():
             lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
         ex = gr.Examples(examples=examples, inputs=[prompt, negative_prompt])