Update app.py

try1 for GPU with state error: moving set_pipe outside of Blocks

app.py

@@ -31,6 +31,17 @@ pipe_inversion.scheduler_inference = scheduler_class.from_config(pipe_inference.
     # image = editor.edit(target_prompt)
     # return image
 
+@spaces.GPU
+def set_pipe(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):
+    config = RunConfig(num_inference_steps=num_inference_steps,
+                       num_inversion_steps=num_inversion_steps,
+                       edit_guidance_scale=edit_guidance_scale,
+                       inversion_max_step=inversion_max_step)
+    return ImageEditorDemo(pipe_inversion, pipe_inference, input_image,
+                           description_prompt, config, device,
+                           [rnri_iterations, rnri_alpha, rnri_lr]), "Input has set!"
+
 def edit(editor, target_prompt):
     if editor is None:
         raise gr.Error("Set inputs before editing.")
@@ -67,17 +78,6 @@ with gr.Blocks(css="style.css") as demo:
     """)
     editor_state = gr.State()
 
-    @spaces.GPU
-    def set_pipe(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):
-        config = RunConfig(num_inference_steps=num_inference_steps,
-                           num_inversion_steps=num_inversion_steps,
-                           edit_guidance_scale=edit_guidance_scale,
-                           inversion_max_step=inversion_max_step)
-        return ImageEditorDemo(pipe_inversion, pipe_inference, input_image,
-                               description_prompt, config, device,
-                               [rnri_iterations, rnri_alpha, rnri_lr]), "Input has set!"
-
     gr.Markdown(f"""running on {power_device}""")
     with gr.Row():
         with gr.Column(elem_id="col-container-1"):