Use cached instance

jhj0517 · Nov 15, 2024 · 97a43e9 · 97a43e9
1 parent b82078a
commit 97a43e9
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/backend/bgm_separation/router.py b/backend/bgm_separation/router.py
@@ -15,6 +15,9 @@
 from ..util.config_loader import load_server_config
 
 
+bgm_separation_router = APIRouter()
+
+
 @functools.lru_cache
 def init_bgm_separation_inferencer() -> 'MusicSeparator':
     config = load_server_config()["bgm_separation"]
@@ -26,15 +29,11 @@ def init_bgm_separation_inferencer() -> 'MusicSeparator':
     return inferencer
 
 
-bgm_separation_router = APIRouter()
-bgm_separation_inferencer = init_bgm_separation_inferencer()
-
-
 async def run_bgm_separation(
     audio: np.ndarray,
     params: BGMSeparationParams
 ) -> Tuple[np.ndarray, np.ndarray]:
-    instrumental, vocal, filepaths = bgm_separation_inferencer.separate(
+    instrumental, vocal, filepaths = init_bgm_separation_inferencer().separate(
         audio=audio,
         model_name=params.model_size,
         device=params.device,