Update app.py

app.py

@@ -19,6 +19,7 @@ import huggingface_hub
 import pillow_avif
 import spaces
 import torch
+import gc
 from huggingface_hub import snapshot_download
 from pillow_heif import register_heif_opener
 
@@ -75,7 +76,11 @@ def prepare_pipeline(model_version, enable_realism, enable_anti_blur):
     pipeline = loaded_pipeline_config['pipeline']
     if pipeline is None or pipeline.model_version != model_version:
         del loaded_pipeline_config['pipeline']
-
+        del pipeline
+        
+        gc.collect()
+        torch.cuda.empty_cache()
+        
         model_path = f'./models/InfiniteYou/infu_flux_v1.0/{model_version}'
         print(f'loading model from {model_path}')