mirror of
https://github.com/Priler/jarvis.git
synced 2026-05-26 23:19:46 +00:00
30 lines
732 B
Python
30 lines
732 B
Python
import vosk
|
|
import sys
|
|
import sounddevice as sd
|
|
import queue
|
|
import json
|
|
|
|
model = vosk.Model("model_small")
|
|
samplerate = 16000
|
|
device = 2
|
|
|
|
q = queue.Queue()
|
|
|
|
|
|
def q_callback(indata, frames, time, status):
|
|
if status:
|
|
print(status, file=sys.stderr)
|
|
q.put(bytes(indata))
|
|
|
|
|
|
def va_listen(callback):
|
|
with sd.RawInputStream(samplerate=samplerate, blocksize=8000, device=device, dtype='int16',
|
|
channels=1, callback=q_callback):
|
|
|
|
rec = vosk.KaldiRecognizer(model, samplerate)
|
|
while True:
|
|
data = q.get()
|
|
if rec.AcceptWaveform(data):
|
|
callback(json.loads(rec.Result())["text"])
|
|
#else:
|
|
# print(rec.PartialResult()) |