diff --git a/src/scripts/py/transcribe.py b/src/scripts/py/transcribe.py index 947b956..9fa7cd1 100644 --- a/src/scripts/py/transcribe.py +++ b/src/scripts/py/transcribe.py @@ -4,6 +4,7 @@ import time import numpy as np import pyaudio import vosk +import openwakeword from openwakeword.model import Model CONFIG = json.loads(sys.argv[1]) @@ -24,6 +25,9 @@ vosk_model = vosk.Model(VOSK_MODEL_PATH) recognizer = vosk.KaldiRecognizer(vosk_model, RATE) recognizer.SetWords(True) +# Download all models (TODO: Import single model) +openwakeword.utils.download_models() + oww = Model(wakeword_models=[WAKEWORD], inference_framework="onnx") audio = pyaudio.PyAudio()