akadmin/openwakeword
1
0
Fork 0
mirror of https://github.com/dscripka/openWakeWord.git synced 2026-08-27 18:17:20 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update detect_from_microphone.py

Browse source
This commit is contained in:
Vojtěch Drábek 2025-01-19 08:31:17 +01:00 • committed by GitHub
parent 7607f95985
commit c69f0f2386
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
examples/detect_from_microphone.py
View file

@ -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 @@ parser.add_argument(
"--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 @@ mic_stream = audio.open(format=FORMAT, channels=CHANNELS, rate=RATE, input=True,
# 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())