From 11fce3abefc2c5d734b02ea870929de48a0f2f8c Mon Sep 17 00:00:00 2001 From: Jaikinator Date: Fri, 30 Jun 2023 18:41:13 +0200 Subject: [PATCH] removed kwargs confusions --- autotranscript/autotranscript.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/autotranscript/autotranscript.py b/autotranscript/autotranscript.py index 6f00888..9f14886 100644 --- a/autotranscript/autotranscript.py +++ b/autotranscript/autotranscript.py @@ -19,8 +19,7 @@ class AutoTranscribe: def __init__(self, whisper_model: Union[bool, str, whisper] = None, dia_model : Union[bool, str, diarisation] = None, - dia_kwargs : dict = {}, - whisper_kwargs : dict = {}) -> None: + **kwargs) -> None: """ AutoTranscribe class @@ -38,16 +37,16 @@ class AutoTranscribe: """ if whisper_model is None: - self.transcriber = Transcriber.load_model("medium", local=True) + self.transcriber = Transcriber.load_model("medium") elif isinstance(whisper_model, str): - self.transcriber = Transcriber.load_model(whisper_model, **whisper_kwargs) + self.transcriber = Transcriber.load_model(whisper_model, **kwargs) else: self.transcriber = whisper_model if dia_model is None: self.diariser = Diariser.load_model() elif isinstance(dia_model, str): - self.diariser = Diariser.load_model(dia_model, **dia_kwargs) + self.diariser = Diariser.load_model(dia_model, **kwargs) else: self.diariser = dia_model