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,