diff --git a/.github/workflows/docker_publish.yml b/.github/workflows/docker_publish.yml index 14d0ca2..e541a8e 100644 --- a/.github/workflows/docker_publish.yml +++ b/.github/workflows/docker_publish.yml @@ -116,6 +116,7 @@ jobs: - ko - ur - te + - hi model: - tiny - base diff --git a/.vscode/settings.json b/.vscode/settings.json index 044ea59..cc6ac9d 100644 --- a/.vscode/settings.json +++ b/.vscode/settings.json @@ -11,6 +11,7 @@ "mpoyraz", "nguyenvulebinh", "saattrupdan", + "theainerd", "torchaudio", "VOXPOPULI", "whisperx", diff --git a/docker-bake.hcl b/docker-bake.hcl index e9f781a..086a9b9 100644 --- a/docker-bake.hcl +++ b/docker-bake.hcl @@ -44,7 +44,8 @@ target "build" { "vi", "ko", "ur", - "te" + "te", + "hi", ] } diff --git a/load_align_model.py b/load_align_model.py index 59a5400..51f6ec9 100644 --- a/load_align_model.py +++ b/load_align_model.py @@ -34,6 +34,7 @@ "ko": "kresnik/wav2vec2-large-xlsr-korean", "ur": "kingabzpro/wav2vec2-large-xls-r-300m-Urdu", "te": "anuragshas/wav2vec2-large-xlsr-53-telugu", + "hi": "theainerd/Wav2Vec2-large-xlsr-hindi", } # From https://github.com/m-bain/whisperX/issues/189#issuecomment-1523392800