Spaces:
Running
Running
jhj0517
commited on
Commit
·
f4609a6
1
Parent(s):
3ce1c87
fixed language code bug
Browse files
modules/faster_whisper_inference.py
CHANGED
|
@@ -263,6 +263,9 @@ class FasterWhisperInference(BaseInterface):
|
|
| 263 |
elapsed time for transcription
|
| 264 |
"""
|
| 265 |
start_time = time.time()
|
|
|
|
|
|
|
|
|
|
| 266 |
segments, info = self.model.transcribe(
|
| 267 |
audio=audio,
|
| 268 |
language=lang,
|
|
|
|
| 263 |
elapsed time for transcription
|
| 264 |
"""
|
| 265 |
start_time = time.time()
|
| 266 |
+
if lang:
|
| 267 |
+
language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
|
| 268 |
+
lang = language_code_dict[lang]
|
| 269 |
segments, info = self.model.transcribe(
|
| 270 |
audio=audio,
|
| 271 |
language=lang,
|