Spaces:
Running
Running
jhj0517
commited on
Commit
·
6726c6a
1
Parent(s):
0baf176
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,
|