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

changed cache default to true

Browse Source
This commit is contained in:
Jaikinator
2023-09-01 14:09:57 +02:00
parent 4aecc63c3a
commit d3e4c2dc75
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
autotranscript/diarisation.py
+3 -3
View File
@@ -152,7 +152,7 @@ class Diariser:
Returns: Returns:
str: The Huggingface token. str: The Huggingface token.
""" """
if os.path.exists(TOKEN_PATH): if os.path.exists(TOKEN_PATH):
with open(TOKEN_PATH, 'r', encoding="utf-8") as file: with open(TOKEN_PATH, 'r', encoding="utf-8") as file:
token = file.read() token = file.read()
@@ -178,7 +178,7 @@ class Diariser:
def load_model(cls, def load_model(cls,
model: str = PYANNOTE_DEFAULT_CONFIG, model: str = PYANNOTE_DEFAULT_CONFIG,
use_auth_token: str = None, use_auth_token: str = None,
cache_token: bool = False, cache_token: bool = True,
cache_dir: Union[Path, str] = PYANNOTE_DEFAULT_PATH, cache_dir: Union[Path, str] = PYANNOTE_DEFAULT_PATH,
hparams_file: Union[str, Path] = None, hparams_file: Union[str, Path] = None,
*args, **kwargs *args, **kwargs
@@ -208,7 +208,7 @@ class Diariser:
if not os.path.exists(model) and use_auth_token is None: if not os.path.exists(model) and use_auth_token is None:
use_auth_token = cls._get_token() use_auth_token = cls._get_token()
model = 'pyannote/speaker-diarization' model = 'pyannote/speaker-diarization'
_model = Pipeline.from_pretrained(model, _model = Pipeline.from_pretrained(model,
use_auth_token = use_auth_token, use_auth_token = use_auth_token,
cache_dir = cache_dir, cache_dir = cache_dir,