Update app.py

app.py

@@ -52,7 +52,7 @@ def generate(image_editor, prompt, neg_prompt, versions):
 
     final_dict = {'background': final_image, 'layers': None, 'composite': final_image}
     
-    return final_dict, image, mask, final_image, [final_image]
+    return final_dict, image, mask, final_image, versions.append(final_image)
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -66,9 +66,9 @@ with gr.Blocks() as demo:
             prompt = gr.Textbox()
             generate_button = gr.Button("Generate")
             with gr.Accordion("Advanced Settings", open=False):
-                neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed, nsfw')
+                neg_prompt = gr.Textbox(label='Negative Prompt', value='ugly, deformed')
         with gr.Column():
-            version_gallery = gr.Gallery(label="Versions")
+            version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain')
             restore_button = gr.Button("Restore Version")