jhj0517 commited on
Commit
0facd17
·
1 Parent(s): 2a0e0ea

refactoring

Browse files
modules/faster_whisper_inference.py CHANGED
@@ -252,14 +252,14 @@ class FasterWhisperInference(BaseInterface):
252
  self.update_model(params.model_size, params.compute_type, progress)
253
 
254
  if params.lang == "Automatic Detection":
255
- lang = None
256
  else:
257
  language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
258
- lang = language_code_dict[params.lang]
259
 
260
  segments, info = self.model.transcribe(
261
  audio=audio,
262
- language=lang,
263
  task="translate" if params.is_translate and self.current_model_size in self.translatable_models else "transcribe",
264
  beam_size=params.beam_size,
265
  log_prob_threshold=params.log_prob_threshold,
 
252
  self.update_model(params.model_size, params.compute_type, progress)
253
 
254
  if params.lang == "Automatic Detection":
255
+ params.lang = None
256
  else:
257
  language_code_dict = {value: key for key, value in whisper.tokenizer.LANGUAGES.items()}
258
+ params.lang = language_code_dict[params.lang]
259
 
260
  segments, info = self.model.transcribe(
261
  audio=audio,
262
+ language=params.lang,
263
  task="translate" if params.is_translate and self.current_model_size in self.translatable_models else "transcribe",
264
  beam_size=params.beam_size,
265
  log_prob_threshold=params.log_prob_threshold,