Set output dir as cache dir

jhj0517 · Nov 26, 2024 · fad840f · fad840f
1 parent 7c26e87
commit fad840f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/backend/routers/transcription/router.py b/backend/routers/transcription/router.py
@@ -11,6 +11,7 @@
 from sqlalchemy.orm import Session
 from datetime import datetime
 from modules.whisper.data_classes import *
+from modules.utils.paths import BACKEND_CACHE_DIR
 from modules.whisper.faster_whisper_inference import FasterWhisperInference
 from backend.common.audio import read_audio
 from backend.common.models import QueueResponse
@@ -28,7 +29,9 @@
 @functools.lru_cache
 def get_pipeline() -> 'FasterWhisperInference':
     config = load_server_config()["whisper"]
-    inferencer = FasterWhisperInference()
+    inferencer = FasterWhisperInference(
+        output_dir=BACKEND_CACHE_DIR
+    )
     inferencer.update_model(
         model_size=config["model_size"],
         compute_type=config["compute_type"]