Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -272,7 +272,7 @@ class WhisperBase(ABC):
|
|
272 |
# Translate the transcribed segments
|
273 |
if translate_output:
|
274 |
self.nllb_inf = NLLBInference()
|
275 |
-
if file_language in
|
276 |
transcribed_segments = self.nllb_inf.translate_text(
|
277 |
input_list_dict=transcribed_segments,
|
278 |
model_size=translate_model,
|
|
|
272 |
# Translate the transcribed segments
|
273 |
if translate_output:
|
274 |
self.nllb_inf = NLLBInference()
|
275 |
+
if file_language in NLLB_AVAILABLE_LANGS.keys():
|
276 |
transcribed_segments = self.nllb_inf.translate_text(
|
277 |
input_list_dict=transcribed_segments,
|
278 |
model_size=translate_model,
|