Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
@@ -341,9 +341,11 @@ class WhisperBase(ABC):
|
|
341 |
*whisper_params,
|
342 |
)
|
343 |
# Define source language
|
344 |
-
source_lang = file_language
|
345 |
-
|
346 |
-
|
|
|
|
|
347 |
|
348 |
# Translate to English using Whisper built-in functionality
|
349 |
transcription_note = ""
|
|
|
341 |
*whisper_params,
|
342 |
)
|
343 |
# Define source language
|
344 |
+
#source_lang = file_language
|
345 |
+
if params.lang == "Automatic Detection" or (params.lang).strip() == "":
|
346 |
+
source_lang = file_language
|
347 |
+
else:
|
348 |
+
source_lang = ((params.lang).strip()).capitalize()
|
349 |
|
350 |
# Translate to English using Whisper built-in functionality
|
351 |
transcription_note = ""
|