Update app.py

app.py

@@ -66,7 +66,7 @@ def generate(image_editor, prompt, neg_prompt, versions):
         final_gallery = versions
         final_gallery.append(final_image)
     
-    return final_dict, final_gallery
+    return final_dict, gr.Gallery(value=final_gallery, visible=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -82,7 +82,7 @@ with gr.Blocks() as demo:
             with gr.Accordion("Advanced Settings", open=False):
                 neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed')
         with gr.Column():
-            version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain')
+            version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain', visible=False)
             restore_button = gr.Button("Restore Version")
             selected = gr.Number(show_label=False, visible=False)