Update app.py

app.py

@@ -67,6 +67,8 @@ def process_image(
                 optimization_steps=optimization_steps,
             ).images[0]
             pipe.vae = pipe.vae.to(torch.float16)
+            torch.cuda.empty_cache()
+            
         return result, f"✅ Inpainting completed with seed {seed}."
     elif mode == "Editing":
         if not edit_prompt:
@@ -94,6 +96,8 @@ def process_image(
                 negative_prompt=prompt,
                 source_steps=max_source_steps,
             ).images[0]
+            torch.cuda.empty_cache()
+            
         return result, f"✅ Editing completed with seed {seed}."
     else:
         return None, f"❌ Invalid mode selected."