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..68c21a1 100644 --- a/server.py +++ b/server.py @@ -28,6 +28,7 @@ def recv_audio(websocket): global clients options = websocket.recv() options = json.loads(options) + print(options) client = ServeClient( websocket, multilingual=options["multilingual"], @@ -55,7 +56,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",