From 922ad18ebcc117765de13692c49f024a09b7bfb9 Mon Sep 17 00:00:00 2001 From: Tijs Zwinkels Date: Wed, 14 Feb 2024 17:29:45 +0100 Subject: [PATCH] Make OpenAI backend work with language autodetect --- whisper_online.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/whisper_online.py b/whisper_online.py index 1bb2a28..8c1916e 100644 --- a/whisper_online.py +++ b/whisper_online.py @@ -152,12 +152,12 @@ class FasterWhisperASR(ASRBase): class OpenaiApiASR(ASRBase): """Uses OpenAI's Whisper API for audio transcription.""" - def __init__(self, lan=None, response_format="verbose_json", temperature=0, logfile=sys.stderr): + def __init__(self, lan=None, temperature=0, logfile=sys.stderr): self.logfile = logfile self.modelname = "whisper-1" - self.language = lan # ISO-639-1 language code - self.response_format = response_format + self.original_language = None if lan == "auto" else lan # ISO-639-1 language code + self.response_format = "verbose_json" self.temperature = temperature self.load_model() @@ -213,8 +213,8 @@ class OpenaiApiASR(ASRBase): "temperature": self.temperature, "timestamp_granularities": ["word", "segment"] } - if self.task != "translate" and self.language: - params["language"] = self.language + if self.task != "translate" and self.original_language: + params["language"] = self.original_language if prompt: params["prompt"] = prompt