Update app.py

app.py

@@ -61,12 +61,11 @@ with gr.Blocks(css="style.css") as demo:
     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 device == 'cuda':
+            torch.cuda.empty_cache()
+            
         if image_editor is not None:
             image_editor = image_editor.to('cpu')
-            if device == 'cuda':
-                torch.cuda.empty_cache()
-                # print("#### torch.cuda.memory_summary(device=None, abbreviated=False):")
-                # print(torch.cuda.memory_summary(device=None, abbreviated=False))
 
         if input_image is None or not description_prompt:
             return None, "Please set all inputs."