From 050555556dcef20acb30f37115d4ad99990eb461 Mon Sep 17 00:00:00 2001 From: "Schmieder, Jacob" Date: Wed, 24 Apr 2024 13:30:00 +0000 Subject: [PATCH] updated transcriber --- scraibe/transcriber.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/scraibe/transcriber.py b/scraibe/transcriber.py index dbb290e..649330b 100644 --- a/scraibe/transcriber.py +++ b/scraibe/transcriber.py @@ -64,14 +64,17 @@ class Transcriber: The class supports various sizes and versions of Whisper models. Please refer to the load_model method for available options. """ - def __init__(self, model: whisper ) -> None: + def __init__(self, model: whisper , model_name: str ) -> None: """ Initialize the Transcriber class with a Whisper model. Args: model (whisper): The Whisper model to use for transcription. """ + self.model = model + + self.model_name = model_name def transcribe(self, audio : Union[str, Tensor, ndarray] , *args, **kwargs) -> str: @@ -156,7 +159,7 @@ class Transcriber: _model = load_model(model, download_root=download_root, device=device, in_memory=in_memory) - return cls(_model) + return cls(_model, model_name=model) @staticmethod def _get_whisper_kwargs(**kwargs) -> dict: @@ -179,4 +182,4 @@ class Transcriber: return whisper_kwargs def __repr__(self) -> str: - return f"Transcriber(model={self.model})" \ No newline at end of file + return f"Transcriber(model_name={self.model_name}, model={self.model})" \ No newline at end of file