Update app.py

app.py

@@ -23,6 +23,7 @@ import tempfile
 import cv2
 import os
 import requests
+import gc
 
 snapshot_download(repo_id="LiuZichen/MagicQuill-models", repo_type="model", local_dir="models")
 # HF_TOKEN = os.environ.get("HF_TOKEN")
@@ -177,6 +178,11 @@ def generate(ckpt_name, total_mask, original_image, add_color_image, add_edge_im
     )
 
     final_image_base64 = tensor_to_base64(final_image)
+    
+    del latent_samples, final_image, lineart_output, color_output
+    gc.collect()
+    torch.cuda.empty_cache()
+    
     return final_image_base64
 
 def generate_image_handler(x, ckpt_name, negative_prompt, fine_edge, grow_size, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):