Update app.py

inserting functions to Blocks

app.py

@@ -31,23 +31,6 @@ 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!"
-
-@spaces.GPU
-def edit(editor, target_prompt):
-    if editor is None:
-        raise gr.Error("Set inputs before editing.")
-    return editor.edit(target_prompt)
-
 # examples = [
 #     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
 #     "An astronaut riding a green horse",
@@ -79,6 +62,23 @@ 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!"
+    
+    @spaces.GPU
+    def edit(editor, target_prompt):
+        if editor is None:
+            raise gr.Error("Set inputs before editing.")
+        return editor.edit(target_prompt)
+
     gr.Markdown(f"""running on {power_device}""")
     with gr.Row():
         with gr.Column(elem_id="col-container-1"):