diff --git a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py
index a6606de4..18f75d8b 100644
--- a/extensions-builtin/sd_forge_svd/scripts/forge_svd.py
+++ b/extensions-builtin/sd_forge_svd/scripts/forge_svd.py
@@ -45,9 +45,11 @@ def on_ui_tabs():
                     refresh_button.click(
                         fn=lambda: gr.update(choices=update_svd_filenames),
                         inputs=[], outputs=filename)
+                    generate_button = gr.Button(value="Generate")
 
             with gr.Column():
-                png_output = gr.Button(value="Save PNG")
+                output_gallery = gr.Gallery(label='Gallery', show_label=False, object_fit='contain',
+                                            visible=True, height=1024, columns=4)
 
     return [(svd_block, "SVD", "svd")]