Update app.py

app.py

@@ -498,7 +498,7 @@ with gr.Blocks(css="style.css") as demo:
                                 max_lines=1, value="", scale=3,
                                 placeholder="Target prompt, DDPM Inversion", info = "Leave blank to have only SEGA concepts guide your edit"
                             )
-                caption_button = gr.Button("Caption Image", scale=1)
+                # caption_button = gr.Button("Caption Image", scale=1)
         
         
 
@@ -625,11 +625,11 @@ with gr.Blocks(css="style.css") as demo:
                                           interactive=True,
                                          info = "Lower the threshold for more effect (e.g. ~0.9 for style transfer)")
 
-    caption_button.click(
-        fn = caption_image,
-        inputs = [input_image],
-        outputs = [tar_prompt]
-    )
+    # caption_button.click(
+    #     fn = caption_image,
+    #     inputs = [input_image],
+    #     outputs = [tar_prompt]
+    # )
     #neg_guidance_1.change(fn = update_label, inputs=[neg_guidance_1], outputs=[add_1])
     #neg_guidance_2.change(fn = update_label, inputs=[neg_guidance_2], outputs=[add_2])
     #neg_guidance_3.change(fn = update_label, inputs=[neg_guidance_3], outputs=[add_3])
@@ -687,7 +687,9 @@ with gr.Blocks(css="style.css") as demo:
     input_image.upload(
         fn = reset_do_inversion,
         outputs = [do_inversion],
-        queue = False).then(fn = update_inversion_progress_visibility, inputs =[input_image,do_inversion],
+        queue = False).then(fn = caption_image,
+        inputs = [input_image],
+        outputs = [tar_prompt]).then(fn = update_inversion_progress_visibility, inputs =[input_image,do_inversion],
                             outputs=[inversion_progress],queue=False).then(
         fn=load_and_invert,
         inputs=[input_image,