admin-valentin/WhisperLiveKit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make OpenAI backend work with language autodetect

Browse source
This commit is contained in:
Tijs Zwinkels 2024-02-14 17:29:45 +01:00
parent f0a24cd5e1
commit 922ad18ebc
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
whisper_online.py
View file

@ -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