feat: Transcriber akzeptiert optionales vorgeladenes WhisperModel

2026-04-12 12:26:20 +02:00
parent e0893917c1
commit 3a580990ea
2 changed files with 22 additions and 5 deletions
@@ -24,11 +24,22 @@ def _model_cache_dir() -> str | None:


 class Transcriber:
-    def __init__(self, model_name: str = "small", compute_type: str = "int8", language: str = "de"):
+    def __init__(
+        self,
+        model_name: str = "small",
+        compute_type: str = "int8",
+        language: str = "de",
+        model: WhisperModel | None = None,
+    ):
        self.language = language
-        logger.info("Lade Whisper-Modell '%s' (compute_type=%s)...", model_name, compute_type)
-        self.model = WhisperModel(model_name, compute_type=compute_type, download_root=_model_cache_dir())
-        logger.info("Modell geladen")
+        if model is not None:
+            self.model = model
+        else:
+            logger.info("Lade Whisper-Modell '%s' (compute_type=%s)...", model_name, compute_type)
+            self.model = WhisperModel(
+                model_name, compute_type=compute_type, download_root=_model_cache_dir()
+            )
+            logger.info("Modell geladen")

    def transcribe(self, audio: np.ndarray) -> str:
        """Transkribiert Audio-Array zu Text."""