diff --git a/server_fastapi.py b/server_fastapi.py
index 517500e..2686680 100644
--- a/server_fastapi.py
+++ b/server_fastapi.py
@@ -2,7 +2,7 @@
 api服务 多版本多模型 fastapi实现
 """
 import logging
-
+import gc
 import utils
 from fastapi import FastAPI, Query
 from fastapi.responses import Response, FileResponse
@@ -139,6 +139,7 @@ class Models:
         # 删除模型
         logger.success(f"卸载模型{model_path}, id = {index}")
         self.models.pop(index)
+        gc.collect()
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
         return index
@@ -291,7 +292,7 @@ if __name__ == "__main__":
 
     def _get_all_models(root_dir: str = "Data", only_unloaded: bool = False):
         result: Dict[str, List[str]] = dict()
-        files = os.listdir(root_dir)
+        files = os.listdir(root_dir) + ["."]
         for file in files:
             if os.path.isdir(os.path.join(root_dir, file)):
                 sub_dir = os.path.join(root_dir, file)