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, image, mask, final_image, final_gallery
+    return final_dict, final_gallery
 
 with gr.Blocks() as demo:
     gr.Markdown("""
@@ -84,16 +84,11 @@ with gr.Blocks() as demo:
         with gr.Column():
             version_gallery = gr.Gallery(label="Versions", type="pil", object_fit='contain')
             restore_button = gr.Button("Restore Version")
-            selected = gr.Number(show_label=False, visible=True)
-            
-        
-    with gr.Row():
-        out1 = gr.Image(format='png')
-        out2 = gr.Image(format='png')
-        out3 = gr.Image(format='png')
+            selected = gr.Number(show_label=False, visible=False)
+    
 
     version_gallery.select(get_select_index, None, selected)
-    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, out1, out2, out3, version_gallery])
+    generate_button.click(fn=generate, inputs=[sketch_pad,prompt, neg_prompt, version_gallery], outputs=[sketch_pad, version_gallery])
     restore_button.click(fn=restore_version, inputs=[selected, version_gallery], outputs=sketch_pad)
 
 demo.launch()