Update app.py

app.py

@@ -60,15 +60,16 @@ with gr.Blocks(css="style.css") as demo:
     @spaces.GPU
     def set_pipe(image_editor, 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 image_editor is not None:
+            image_editor = image_editor.to('cpu')
+
         if input_image is None or not description_prompt:
             return None, "Please set all inputs."
 
         print('### description_prompt ', description_prompt)
         print('### input_image.size ', input_image.size)
         
-        if image_editor is not None:
-            image_editor = image_editor.to('cpu')
-
         if isinstance(num_inference_steps, str): num_inference_steps = int(num_inference_steps) 
         if isinstance(num_inversion_steps, str): num_inversion_steps = int(num_inversion_steps) 
         if isinstance(edit_guidance_scale, str): edit_guidance_scale = float(edit_guidance_scale)