Update app.py

app.py

@@ -59,11 +59,15 @@ with gr.Blocks(css="style.css") as demo:
                     
         scheduler_class = MyEulerAncestralDiscreteScheduler
         
+        print('\n################## 1')
         pipe_inversion = SDXLDDIMPipeline.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to(device)
+        print('\n################## 2')
         pipe_inference = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", use_safetensors=True).to(device)
+        print('\n################## 3')
         pipe_inference.scheduler = scheduler_class.from_config(pipe_inference.scheduler.config)
         pipe_inversion.scheduler = scheduler_class.from_config(pipe_inversion.scheduler.config)
         pipe_inversion.scheduler_inference = scheduler_class.from_config(pipe_inference.scheduler.config)
+        print('\n################## 4')
 
 
         config = RunConfig(num_inference_steps=num_inference_steps,
@@ -71,8 +75,9 @@ with gr.Blocks(css="style.css") as demo:
                            edit_guidance_scale=edit_guidance_scale,
                            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]).to('cpu')
+                               description_prompt, config, 'cpu',
+                               [rnri_iterations, rnri_alpha, rnri_lr])
+        print('\n################## 5')
         return image_editor, "Input has set!"
     
     @spaces.GPU