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,
|