1 files changed, 2 insertions, 0 deletions
diff --git a/webui.py b/webui.py
index 7e0b3296..e71a217c 100644
--- a/webui.py
+++ b/webui.py
@@ -18,9 +18,11 @@ import modules.shared as shared
 import modules.swinir_model as swinir
 import modules.txt2img
 import modules.ui
+from modules import modelloader
 from modules.paths import script_path
 from modules.shared import cmd_opts
 
+modelloader.cleanup_models()
 codeformer.setup_model(cmd_opts.codeformer_models_path)
 gfpgan.setup_model(cmd_opts.gfpgan_models_path)
 shared.face_restorers.append(modules.face_restoration.FaceRestoration())