set lang to en if not multilingual

2023-07-31 14:40:38 +08:00
parent caac4ac47b
commit f20c4153d4
2 changed files with 3 additions and 2 deletions
@@ -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)

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