Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
|
@@ -259,8 +259,8 @@ class WhisperBase(ABC):
|
|
| 259 |
#files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
|
| 260 |
|
| 261 |
# Translate the trandscribed segments
|
| 262 |
-
|
| 263 |
-
test =
|
| 264 |
input_list_dict=transcribed_segments,
|
| 265 |
model_size="facebook/nllb-200-distilled-600M",
|
| 266 |
src_lang=file_language,
|
|
|
|
| 259 |
#files_info[file_name] = {"subtitle": subtitle, "time_for_task": time_for_task, "path": file_path, "lang": file_language, "lang_prob": file_lang_probs, "input_source_file": (file_name+file_ext)}
|
| 260 |
|
| 261 |
# Translate the trandscribed segments
|
| 262 |
+
TranslateInstance = TranslationBase()
|
| 263 |
+
test = TranslateInstance.translate_text(
|
| 264 |
input_list_dict=transcribed_segments,
|
| 265 |
model_size="facebook/nllb-200-distilled-600M",
|
| 266 |
src_lang=file_language,
|