Update app.py

moving editor to cpu before forwarding to State.

app.py

@@ -72,14 +72,18 @@ with gr.Blocks(css="style.css") as demo:
                            inversion_max_step=inversion_max_step)
         image_editor = ImageEditorDemo(pipe_inversion, pipe_inference, input_image,
                                description_prompt, config, device,
-                               [rnri_iterations, rnri_alpha, rnri_lr])
+                               [rnri_iterations, rnri_alpha, rnri_lr]).to('cpu')
         return image_editor, "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)
+        if device == "cuda":
+            image = editor.to(device).edit(target_prompt)
+        else:
+            image = editor.edit(target_prompt)
+        return image
 
     gr.Markdown(f"""running on {power_device}""")
     with gr.Row():