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 = ""
|