Update app.py

app.py

@@ -62,11 +62,11 @@ with gr.Blocks(css="style.css") as demo:
                  num_inversion_steps=4, inversion_max_step=0.6, rnri_iterations=2, rnri_alpha=0.1, rnri_lr=0.2):
         
         if device == 'cuda':
+            if image_editor is not None:
+                image_editor = image_editor.to('cpu')
+
             torch.cuda.empty_cache()
             
-        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."