From c69f0f23867c99965ede64cd7912d744f97c91f5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Vojt=C4=9Bch=20Dr=C3=A1bek?= Date: Sun, 19 Jan 2025 08:31:17 +0100 Subject: [PATCH] Update detect_from_microphone.py --- examples/detect_from_microphone.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/examples/detect_from_microphone.py b/examples/detect_from_microphone.py index 6e69c92..0811de0 100644 --- a/examples/detect_from_microphone.py +++ b/examples/detect_from_microphone.py @@ -13,9 +13,10 @@ # limitations under the License. # Imports +import os import pyaudio import numpy as np -from openwakeword.model import Model +import openwakeword import argparse # Parse input arguments @@ -38,7 +39,7 @@ "--inference_framework", help="The inference framework to use (either 'onnx' or 'tflite'", type=str, - default='tflite', + default='onnx' if os.name == 'nt' else 'tflite', required=False ) @@ -54,9 +55,10 @@ # Load pre-trained openwakeword models if args.model_path != "": - owwModel = Model(wakeword_models=[args.model_path], inference_framework=args.inference_framework) + owwModel = openwakeword.model.Model(wakeword_models=[args.model_path], inference_framework=args.inference_framework) else: - owwModel = Model(inference_framework=args.inference_framework) + openwakeword.utils.download_models() + owwModel = openwakeword.model.Model(inference_framework=args.inference_framework) n_models = len(owwModel.models.keys())