mirror of
https://github.com/wassname/BitLit_test1.git
synced 2026-06-28 01:58:06 +08:00
wassname
70 lines
1.9 KiB
Python
70 lines
1.9 KiB
Python
#### RUNNING THE MODEL
|
|
#Michels-MacBook-Pro:~ ShebMichel$ cd documents/pmlg/wake/decoder
|
|
#Michels-MacBook-Pro:decoder ShebMichel$ python demo.py resources/HiBitLit.pmdl
|
|
import sys, os
|
|
import sys
|
|
|
|
import snowboydecoder
|
|
import signal
|
|
#import BitLit_main ## MAIN PROGRAM
|
|
|
|
####
|
|
from gtts import gTTS ## Packages for Text to voice
|
|
import speech_recognition as sr ## Packages for voice recognizer
|
|
import tensorflow as tf
|
|
tf.enable_eager_execution()
|
|
from tensorflow.keras.layers import Embedding, GRU, Dense
|
|
import numpy as np
|
|
import re
|
|
from textblob import TextBlob
|
|
import random
|
|
import poem_generator ## POEM GENERATOR IMPORT
|
|
from poem_generator import*
|
|
import time
|
|
####
|
|
t0=time.time() ## Time counter
|
|
interrupted = False
|
|
|
|
|
|
def signal_handler(signal, frame):
|
|
global interrupted
|
|
interrupted = True
|
|
|
|
|
|
def interrupt_callback():
|
|
global interrupted
|
|
return interrupted
|
|
|
|
if len(sys.argv) == 1:
|
|
print("Error: need to specify model name")
|
|
print("Usage: python demo.py your.model")
|
|
sys.exit(-1)
|
|
|
|
model = sys.argv[1]
|
|
|
|
# capture SIGINT signal, e.g., Ctrl+C
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
|
|
detector = snowboydecoder.HotwordDetector(model, sensitivity=0.5)
|
|
print('Listening... Press Ctrl+C to exit')
|
|
|
|
# main loop
|
|
detector.start(detected_callback=snowboydecoder.run,
|
|
interrupt_check=interrupt_callback,
|
|
sleep_time=0.01)
|
|
##detector.start(detected_callback=snowboydecoder.play_audio_file,
|
|
## interrupt_check=interrupt_callback,
|
|
## sleep_time=0.00)
|
|
##detector.start(detected_callback= os.system('python BitLit_main.py'),
|
|
## interrupt_check=interrupt_callback,
|
|
## sleep_time=0.00) ##0.03
|
|
#import BitLit_main
|
|
##detector.start(detected_callback=BitLit_main,
|
|
## interrupt_check=interrupt_callback,
|
|
## sleep_time=0.03)
|
|
|
|
detector.terminate()
|
|
t1 =time.time()
|
|
total=t1-t0
|
|
print(('Time spent is about:', np.round(total), 'seconds'))
|