DEV/alphacep_vosk-server
1
0
Fork 0
mirror of https://github.com/alphacep/vosk-server.git synced 2026-01-09 07:40:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add model argument

Browse Source
This commit is contained in:
Nickolay Shmyrev 2023-03-05 23:32:14 +01:00
parent c957776871
commit d8278fbd66
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
websocket-gpu-batch/asr_server_gpu.py
View File

@ -78,10 +78,14 @@ async def start():
args.interface = os.environ.get('VOSK_SERVER_INTERFACE', '0.0.0.0')
args.port = int(os.environ.get('VOSK_SERVER_PORT', 2700))
args.sample_rate = float(os.environ.get('VOSK_SAMPLE_RATE', 8000))
args.model_path = os.environ.get('VOSK_MODEL_PATH', 'model')
if len(sys.argv) > 1:
args.model_path = sys.argv[1]
GpuInit()
model = BatchModel()
model = BatchModel(args.model_path)
async with websockets.serve(recognize, args.interface, args.port):
await asyncio.Future()