Update app.py

reconstruction logic change

app.py

@@ -72,6 +72,7 @@ def sample(zs, wts, prompt_tar="", cfg_scale_tar=15, skip=36, eta = 1):
 
 
 def reconstruct(tar_prompt,
+                image_caption,
                 tar_cfg_scale,
                 skip,
                 wts, zs,
@@ -85,6 +86,8 @@ def reconstruct(tar_prompt,
 
     else:
       if do_reconstruction:
+          if image_caption.lower() == tar_prompt.lower(): # if image caption was not changed, run actual reconstruction
+             tar_prompt = ""
           reconstruction_img = sample(zs.value, wts.value, prompt_tar=tar_prompt, skip=skip, cfg_scale_tar=tar_cfg_scale)
           reconstruction = gr.State(value=reconstruction_img)
           do_reconstruction = False
@@ -800,6 +803,7 @@ with gr.Blocks(css="style.css") as demo:
 
     reconstruct_button.click(lambda: ddpm_edited_image.update(visible=True), outputs=[ddpm_edited_image]).then(fn = reconstruct,
                 inputs = [tar_prompt,
+                image_caption,
                 tar_cfg_scale,
                 skip,
                 wts, zs,