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

added deprecated warning

Browse Source
This commit is contained in:
Jaikinator
2023-06-12 11:50:20 +02:00
parent 7aa2ed667f
commit ca42d631cd
1 changed files with 17 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
autotranscript/misc.py
+17 -19
View File
@@ -3,20 +3,14 @@ from pyannote.audio import Pipeline
from whisper import Whisper, load_model
import os
import glob
from warnings import warn
def get_whisper_default_path() -> str:
"""
Get default path for whisper models
WHISPER_DEFAULT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)),
"models", "whisper")
Returns
-------
str
path
"""
_path = os.path.dirname(os.path.dirname(__file__))
return os.path.join(_path, "models", "whisper")
WHISPER_DEFAULT_PATH = get_whisper_default_path()
PYANNOTE_DEFAULT_PATH = os.path.join(os.path.dirname(os.path.dirname(__file__)),
"models", "pyannote",
"speaker_diarization", "config.yaml")
def load_whisper_model(model: str ="medium", local : bool = False, download_root: str = WHISPER_DEFAULT_PATH) -> Whisper:
"""
@@ -52,9 +46,9 @@ def load_whisper_model(model: str ="medium", local : bool = False, download_root
-------
Whisper Object
"""
warn("load_whisper_model is deprecated. Use Transcriptor.load_model() instead.", DeprecationWarning)
if local:
available_models = [os.path.basename(x) for x in glob.glob(os.path.join(WHISPER_DEFAULT_PATH, "*"))]
available_models = [os.path.basename(x) for x in glob.glob(os.path.join(download_root, "*"))]
for i, module in enumerate(available_models):
available_models[i] = module.split(".")[0]
@@ -62,9 +56,12 @@ def load_whisper_model(model: str ="medium", local : bool = False, download_root
if model not in available_models:
raise RuntimeError("Model not found. Consider downloading the model first. By deactivating the local flag, the model will be downloaded automatically.")
return load_model(model, download_root=WHISPER_DEFAULT_PATH)
return load_model(model, download_root=download_root)
def load_pyannote_model(model: str, token: str = "", local : bool = True) -> Pipeline:
def load_pyannote_model(model: str = PYANNOTE_DEFAULT_PATH,
token: str = "",
local : bool = True,
*args, **kwargs) -> Pipeline:
"""
Load modules from pyannote
@@ -72,6 +69,7 @@ def load_pyannote_model(model: str, token: str = "", local : bool = True) -> Pip
----------
model : str
pyannote model
default: /models/pyannote/speaker_diarization/config.yaml
token : str
HUGGINGFACE_TOKEN
local : bool
@@ -81,8 +79,8 @@ def load_pyannote_model(model: str, token: str = "", local : bool = True) -> Pip
-------
Pipeline Object
"""
warn("load_pyannote_model is deprecated. Use Diarisation.load_model() instead.", DeprecationWarning)
if local:
return Pipeline.from_pretrained(model)
return Pipeline.from_pretrained(model,*args, **kwargs)
else:
return Pipeline.from_pretrained(model, use_auth_token = token)
return Pipeline.from_pretrained(model, use_auth_token = token, *args, **kwargs)