cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set lang to en if not multilingual

Browse Source
This commit is contained in:
makaveli10
2023-07-31 14:40:38 +08:00
parent caac4ac47b
commit f20c4153d4
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
client.py
+1 -1
View File
@@ -251,7 +251,7 @@ if __name__=="__main__":
opt = parser.parse_args() opt = parser.parse_args()
print(opt) print(opt)
multilingual=opt.multilingual, multilingual=opt.multilingual,
language = opt.language, language = opt.language if opt.multilingual else "en",
task = opt.task task = opt.task
c = Client(host=opt.host, port=opt.port) c = Client(host=opt.host, port=opt.port)
server.py
+2 -1
View File
@@ -28,6 +28,7 @@ def recv_audio(websocket):
global clients global clients
options = websocket.recv() options = websocket.recv()
options = json.loads(options) options = json.loads(options)
print(options)
client = ServeClient( client = ServeClient(
websocket, websocket,
multilingual=options["multilingual"], multilingual=options["multilingual"],
@@ -55,7 +56,7 @@ class ServeClient:
def __init__(self, websocket, task="transcribe", device=None, multilingual=False, language=None): def __init__(self, websocket, task="transcribe", device=None, multilingual=False, language=None):
self.data = b"" self.data = b""
self.frames = b"" self.frames = b""
self.language = language self.language = language if multilingual else "en"
self.task = task self.task = task
self.transcriber = WhisperModel( self.transcriber = WhisperModel(
"small" if multilingual else "small.en", "small" if multilingual else "small.en",