Skip to content

Commit

Permalink
Fixed log message, updated timeout
Browse files Browse the repository at this point in the history
When translating, the wrong output language was written to the log, but whisper output was not affected.
Also, changed timeout from 5 to passed timeout value in self.timeout as per recent Discord discussion.
  • Loading branch information
JaiZed committed Dec 27, 2023
1 parent dafa553 commit 4a82c4f
Showing 1 changed file with 10 additions and 5 deletions.
15 changes: 10 additions & 5 deletions libs/subliminal_patch/providers/whisperai.py
Original file line number Diff line number Diff line change
Expand Up @@ -256,7 +256,7 @@ def detect_language(self, path) -> Language:
r = self.session.post(f"{self.endpoint}/detect-language",
params={'encode': 'false'},
files={'audio_file': out},
timeout=(5, self.timeout))
timeout=(self.timeout, self.timeout))

logger.debug(f"Whisper detected language of {path} as {r.json()['detected_language']}")

Expand Down Expand Up @@ -321,16 +321,21 @@ def download_subtitle(self, subtitle: WhisperAISubtitle):
subtitle.content = None
return

logger.info(f'Starting WhisperAI {subtitle.task} to {language_from_alpha3(subtitle.audio_language)} for {subtitle.video.original_path}')
if subtitle.task == "transcribe":
output_language = subtitle.audio_language
else:
output_language = "eng"

logger.info(f'Starting WhisperAI {subtitle.task} to {language_from_alpha3(output_language)} for {subtitle.video.original_path}')
startTime = time.time()

r = self.session.post(f"{self.endpoint}/asr",
params={'task': subtitle.task, 'language': whisper_get_language_reverse(subtitle.audio_language), 'output': 'srt', 'encode': 'false'},
params={'task': subtitle.task, 'language': whisper_get_language_reverse(output_language), 'output': 'srt', 'encode': 'false'},
files={'audio_file': out},
timeout=(5, self.timeout))
timeout=(self.timeout, self.timeout))

endTime = time.time()
elapsedTime = timedelta(seconds=round(endTime - startTime))
logger.info(f'Completed WhisperAI {subtitle.task} to {language_from_alpha3(subtitle.audio_language)} in {elapsedTime} for {subtitle.video.original_path}')
logger.info(f'Completed WhisperAI {subtitle.task} to {language_from_alpha3(output_language)} in {elapsedTime} for {subtitle.video.original_path}')

subtitle.content = r.content

0 comments on commit 4a82c4f

Please sign in to comment.