Update modules/whisper/whisper_base.py
Browse files
modules/whisper/whisper_base.py
CHANGED
|
@@ -338,10 +338,10 @@ class WhisperBase(ABC):
|
|
| 338 |
total_info += f'Input file:\t\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
|
| 339 |
|
| 340 |
if params.is_translate:
|
| 341 |
-
total_info += f'Translation:\t{info["transcription"]}\n\t
|
| 342 |
|
| 343 |
if translate_output:
|
| 344 |
-
total_info += f'Translation:\t{info["translation"]}\n\t
|
| 345 |
|
| 346 |
time_end = datetime.now()
|
| 347 |
total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
|
|
@@ -387,7 +387,7 @@ class WhisperBase(ABC):
|
|
| 387 |
Output file path to return to gr.Files()
|
| 388 |
"""
|
| 389 |
try:
|
| 390 |
-
progress(0, desc="Loading Audio
|
| 391 |
transcribed_segments, time_for_task = self.run(
|
| 392 |
mic_audio,
|
| 393 |
progress,
|
|
@@ -442,7 +442,7 @@ class WhisperBase(ABC):
|
|
| 442 |
Output file path to return to gr.Files()
|
| 443 |
"""
|
| 444 |
try:
|
| 445 |
-
progress(0, desc="Loading Audio from Youtube
|
| 446 |
yt = get_ytdata(youtube_link)
|
| 447 |
audio = get_ytaudio(yt)
|
| 448 |
|
|
|
|
| 338 |
total_info += f'Input file:\t\t{info["input_source_file"]}\nLanguage:\t{info["lang"]} (probability {info["lang_prob"]}%)\n'
|
| 339 |
|
| 340 |
if params.is_translate:
|
| 341 |
+
total_info += f'Translation:\t{info["transcription"]}\n\t⤷ handled by OpenAI Whisper\n'
|
| 342 |
|
| 343 |
if translate_output:
|
| 344 |
+
total_info += f'Translation:\t{info["translation"]}\n\t⤷ handled by Facebook NLLB\n'
|
| 345 |
|
| 346 |
time_end = datetime.now()
|
| 347 |
total_info += f"\nTotal processing time: {self.format_time((time_end-time_start).total_seconds())}"
|
|
|
|
| 387 |
Output file path to return to gr.Files()
|
| 388 |
"""
|
| 389 |
try:
|
| 390 |
+
progress(0, desc="Loading Audio...")
|
| 391 |
transcribed_segments, time_for_task = self.run(
|
| 392 |
mic_audio,
|
| 393 |
progress,
|
|
|
|
| 442 |
Output file path to return to gr.Files()
|
| 443 |
"""
|
| 444 |
try:
|
| 445 |
+
progress(0, desc="Loading Audio from Youtube...")
|
| 446 |
yt = get_ytdata(youtube_link)
|
| 447 |
audio = get_ytaudio(yt)
|
| 448 |
|