Update app.py

app.py

@@ -27,7 +27,10 @@ def divisible_by_8(image):
     return resized_image
 
 
-def generate(image_editor, prompt, neg_prompt):
+def generate(image_editor, prompt, neg_prompt, versions):
+    print('versions')
+    print(versions)
+    
     image = image_editor['background'].convert('RGB')
     image.thumbnail((1024, 1024))
     image = divisible_by_8(image)
@@ -47,7 +50,7 @@ def generate(image_editor, prompt, neg_prompt):
     
     final_image = final_image.crop((0, 0, original_image_size[0], original_image_size[1]))
     
-    return image_editor, image, mask, final_image
+    return image_editor, image, mask, final_image, [final_image]
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -72,6 +75,6 @@ with gr.Blocks() as demo:
         out2 = gr.Image(format='png')
         out3 = gr.Image(format='png')
         
-    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt], outputs=[sketch_pad, out1, out2, out3])
+    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, out1, out2, out3, version_gallery])
 
 demo.launch()