Skip to content

Commit

Permalink
feat/accept model from path
Browse files Browse the repository at this point in the history
  • Loading branch information
JarbasAl committed Aug 1, 2024
1 parent 34f3598 commit 1b6ad49
Showing 1 changed file with 11 additions and 7 deletions.
18 changes: 11 additions & 7 deletions neon_stt_plugin_nemo/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,18 +45,22 @@ def __init__(self, config: dict = None):

self._engines = {}
self.cache_engines = self.config.get("cache", True)
self.default_model = self.config.get("model", None)
if self.cache_engines:
self._init_model(self.lang)
self._init_model()

def _init_model(self, language) -> Model:
def _init_model(self, language=None) -> Model:
# OVOS uses 'en-us' so this hacks that to work until dialects are supported
language = (language or self.lang).split('-')[0]
if language not in self._engines:
model = Model(language)
lang = (language or self.lang).split('-')[0]
if lang not in self._engines:
if self.default_model and language is None:
model = Model(lang=lang, model_folder=self.default_model)
else:
model = Model(lang=lang)
if self.cache_engines:
self._engines[language] = model
self._engines[lang] = model
else:
model = self._engines[language]
model = self._engines[lang]

return model

Expand Down

0 comments on commit 1b6ad49

Please sign in to comment.