Update app.py

app.py

@@ -132,6 +132,9 @@ def invert_and_reconstruct(
     inversion_map['wts'] = wts
     
     return pure_ddpm_out
+
+def reset():
+    inversion_map.clear()
     
 def edit(input_image, 
                     src_prompt ="", 
@@ -316,6 +319,10 @@ with gr.Blocks() as demo:
         outputs=[sega_edited_image],
     )
 
+    input_image.change(
+        fn = reset
+    )
+
     gr.Examples(
         label='Examples', 
         examples=get_example(),