Skip to content

Commit

Permalink
more languages
Browse files Browse the repository at this point in the history
  • Loading branch information
Jemoka committed Jun 8, 2024
1 parent e3de0c2 commit 1004e17
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 3 deletions.
2 changes: 2 additions & 0 deletions batchalign/pipelines/asr/whisper.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@ def __init__(self, model=None, lang="eng"):
language = pycountry.languages.get(alpha_3=lang).name
if language == "Yue Chinese":
language = "Cantonese"
if "greek" in language.lower():
language = "Greek"

self.__whisper = WhisperASRModel(model, base=base, language=language)
self.__lang = lang
Expand Down
2 changes: 1 addition & 1 deletion batchalign/pipelines/dispatch.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,7 +73,7 @@ def dispatch_pipeline(pkg_str, lang, num_speakers=None, **arg_overrides):
packages.append("disfluency")
if "retracing" not in packages:
packages.append("retracing")
if "utterance" not in packages and resolve("utterance", lang) == None and lang not in ["heb", "fra", "ara", "nld"]:
if "utterance" not in packages and resolve("utterance", lang) == None and lang not in ["heb", "fra", "ara", "nld", "hun", "eus", "grc", "ell"]:
packages.append("utterance")
if "fa" in packages:
if "utr" not in packages:
Expand Down
4 changes: 2 additions & 2 deletions batchalign/version
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
0.7.3-beta.1
0.7.3-beta.2
June 7, 2024
bug fixes
more languages

0 comments on commit 1004e17

Please sign in to comment.