minor bug fix

app.py

@@ -90,7 +90,7 @@ with gr.Blocks(css="style.css") as demo:
 
 
     @spaces.GPU
-    def set_pipe(inversion_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
+    def set_pipe(inversion_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale, num_inference_steps=4,
                  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
 
         if input_image is None or not description_prompt:
@@ -135,7 +135,7 @@ with gr.Blocks(css="style.css") as demo:
 
 
     @spaces.GPU
-    def edit(inversion_state, target_prompt):
+    def edit(inversion_state, pipe_inference, target_prompt):
         if inversion_state is None:
             raise gr.Error("Set inputs before editing.")
         # if device == "cuda":
@@ -245,42 +245,42 @@ with gr.Blocks(css="style.css") as demo:
     gr.Markdown(f"""Disclaimer: Performance may be inferior to the reported in the paper due to hardware limitation.
     """)
     input_image.change(set_pipe,
-                       inputs=[editor_state, input_image, description_prompt, edit_guidance_scale, num_inference_steps,
+                       inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale, num_inference_steps,
                                num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha, rnri_lr],
                        outputs=[editor_state, is_set_text])
 
-    description_prompt.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    description_prompt.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                                 num_inference_steps,
                                                 num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                                 rnri_lr],
                               outputs=[editor_state, is_set_text])
 
-    edit_guidance_scale.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    edit_guidance_scale.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                                  num_inference_steps,
                                                  num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                                  rnri_lr],
                                outputs=[editor_state, is_set_text])
-    num_inference_steps.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    num_inference_steps.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                                  num_inference_steps,
                                                  num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                                  rnri_lr],
                                outputs=[editor_state, is_set_text])
-    inversion_max_step.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    inversion_max_step.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                                 num_inference_steps,
                                                 num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                                 rnri_lr],
                               outputs=[editor_state, is_set_text])
-    rnri_iterations.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    rnri_iterations.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                              num_inference_steps,
                                              num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                              rnri_lr],
                            outputs=[editor_state, is_set_text])
-    rnri_alpha.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    rnri_alpha.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                         num_inference_steps,
                                         num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                         rnri_lr],
                       outputs=[editor_state, is_set_text])
-    rnri_lr.change(set_pipe, inputs=[editor_state, input_image, description_prompt, edit_guidance_scale,
+    rnri_lr.change(set_pipe, inputs=[editor_state, pipe_inversion, input_image, description_prompt, edit_guidance_scale,
                                      num_inference_steps,
                                      num_inference_steps, inversion_max_step, rnri_iterations, rnri_alpha,
                                      rnri_lr],
@@ -295,7 +295,7 @@ with gr.Blocks(css="style.css") as demo:
 
     run_button.click(
         fn=edit,
-        inputs=[editor_state, target_prompt],
+        inputs=[editor_state, pipe_inference, target_prompt],
         outputs=[result]
     )