Update app.py

app.py

@@ -193,7 +193,7 @@ def edit(input_image,
                         num_images_per_prompt=1,  
                         num_inference_steps=steps, 
                         use_ddpm=True,  wts=wts.value, zs=zs.value[skip:], **editing_args)
-    return sega_out.images[0], True
+    return sega_out.images[0]
 
 
 
@@ -403,7 +403,7 @@ with gr.Blocks(css='style.css') as demo:
                 threshold_1, threshold_2, threshold_3
 
         ],
-        outputs=[sega_edited_image, is_show_reconstruction],     
+        outputs=[sega_edited_image],     
     ).success( 
         fn = show_reconstruction_button,
         outputs = [reconstruct_button]