diff --git a/Audio-Transcription-Chrome/options.js b/Audio-Transcription-Chrome/options.js index 316d5b6..9f685b7 100644 --- a/Audio-Transcription-Chrome/options.js +++ b/Audio-Transcription-Chrome/options.js @@ -82,6 +82,9 @@ async function startRecord(option) { const socket = new WebSocket(`ws://${option.host}:${option.port}/`); let isServerReady = false; let language = option.language; + if (language === null && !option.multilingual) { + language = 'en'; + } socket.onopen = function(e) { socket.send( JSON.stringify({ diff --git a/Audio-Transcription-Firefox/content.js b/Audio-Transcription-Firefox/content.js index 5a6391e..d849b0d 100644 --- a/Audio-Transcription-Firefox/content.js +++ b/Audio-Transcription-Firefox/content.js @@ -42,6 +42,9 @@ function resampleTo16kHZ(audioData, origSampleRate = 44100) { function startRecording(data) { socket = new WebSocket(`ws://${data.host}:${data.port}/`); language = data.language; + if (language === null && !data.useMultilingual) { + language = 'en'; + } socket.onopen = function(e) { socket.send( JSON.stringify({ diff --git a/client.py b/client.py index f912beb..44fe89c 100644 --- a/client.py +++ b/client.py @@ -251,7 +251,7 @@ if __name__=="__main__": opt = parser.parse_args() print(opt) multilingual=opt.multilingual, - language = opt.language, + language = opt.language if opt.multilingual else "en", task = opt.task c = Client(host=opt.host, port=opt.port) diff --git a/server.py b/server.py index 9879b9f..d675ef1 100644 --- a/server.py +++ b/server.py @@ -55,7 +55,7 @@ class ServeClient: def __init__(self, websocket, task="transcribe", device=None, multilingual=False, language=None): self.data = b"" self.frames = b"" - self.language = language + self.language = language if multilingual else "en" self.task = task self.transcriber = WhisperModel( "small" if multilingual else "small.en", diff --git a/transcriber.py b/transcriber.py index e4404e3..ec6e28c 100644 --- a/transcriber.py +++ b/transcriber.py @@ -288,6 +288,7 @@ class WhisperModel: language, language_probability, ) + return language, language_probability else: language_probability = 1