diff --git a/modules/ui_image_generation.py b/modules/ui_image_generation.py
index b71c9f46..435a0300 100644
--- a/modules/ui_image_generation.py
+++ b/modules/ui_image_generation.py
@@ -153,7 +153,7 @@ def create_ui():
             with gr.TabItem("Gallery"):
                 with gr.Row():
                     shared.gradio['image_refresh_history'] = gr.Button("🔄 Refresh Gallery", elem_classes="refresh-button")
-                shared.gradio['image_history_gallery'] = gr.Gallery(label="History", show_label=False, columns=6, object_fit="cover", height="auto", allow_preview=True, elem_id="image-history-gallery")
+                shared.gradio['image_history_gallery'] = gr.Gallery(value=lambda : get_history_images(), label="History", show_label=False, columns=6, object_fit="cover", height="auto", allow_preview=True, elem_id="image-history-gallery")
 
             # TAB 3: MODEL
             with gr.TabItem("Model"):