Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -259,7 +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 |
input_list_dict=transcribed_segments,
|
264 |
model_size="facebook/nllb-200-distilled-600M",
|
265 |
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,
|