admin/scribe
1
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

removed kwargs confusions

Browse Source
This commit is contained in:
Jaikinator
2023-06-30 18:41:13 +02:00
parent de3a6cd4d1
commit 11fce3abef
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
autotranscript/autotranscript.py
+4 -5
View File
@@ -19,8 +19,7 @@ class AutoTranscribe:
def __init__(self, def __init__(self,
whisper_model: Union[bool, str, whisper] = None, whisper_model: Union[bool, str, whisper] = None,
dia_model : Union[bool, str, diarisation] = None, dia_model : Union[bool, str, diarisation] = None,
dia_kwargs : dict = {}, **kwargs) -> None:
whisper_kwargs : dict = {}) -> None:
""" """
AutoTranscribe class AutoTranscribe class
@@ -38,16 +37,16 @@ class AutoTranscribe:
""" """
if whisper_model is None: if whisper_model is None:
self.transcriber = Transcriber.load_model("medium", local=True) self.transcriber = Transcriber.load_model("medium")
elif isinstance(whisper_model, str): elif isinstance(whisper_model, str):
self.transcriber = Transcriber.load_model(whisper_model, **whisper_kwargs) self.transcriber = Transcriber.load_model(whisper_model, **kwargs)
else: else:
self.transcriber = whisper_model self.transcriber = whisper_model
if dia_model is None: if dia_model is None:
self.diariser = Diariser.load_model() self.diariser = Diariser.load_model()
elif isinstance(dia_model, str): elif isinstance(dia_model, str):
self.diariser = Diariser.load_model(dia_model, **dia_kwargs) self.diariser = Diariser.load_model(dia_model, **kwargs)
else: else:
self.diariser = dia_model self.diariser = dia_model