mirror of
https://github.com/wassname/BitLit_test1.git
synced 2026-06-27 22:05:08 +08:00
wassname
61 lines
1.5 KiB
Python
61 lines
1.5 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 BitLit_main
|
|
# 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')
|
|
|
|
|
|
detector.start(detected_callback=BitLit_main.generate_poem,
|
|
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'))
|