o
    SQ8j|  ã                   @   sœ  d dl Z d dlmZ d dlmZ d dlmZ g d¢Ze j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d	¡d
dœdœZde j 	e j 
e j e¡¡d¡ddœiZe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœe j 	e j 
e j e¡¡d¡ddœdœZddddd d!d"d#œiZd'd%d&„ZdS )(é    N)ÚModel)ÚVAD)Útrain_custom_verifier)r   r   r   z'resources/models/embedding_model.tflitezXhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/embedding_model.tflite)Ú
model_pathÚdownload_urlz&resources/models/melspectrogram.tflitezWhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/melspectrogram.tflite)Ú	embeddingÚmelspectrogramÚ
silero_vadz resources/models/silero_vad.onnxzQhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/silero_vad.onnxz"resources/models/alexa_v0.1.tflitezShttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/alexa_v0.1.tflitez(resources/models/hey_mycroft_v0.1.tflitezYhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/hey_mycroft_v0.1.tflitez'resources/models/hey_jarvis_v0.1.tflitezXhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/hey_jarvis_v0.1.tflitez(resources/models/hey_rhasspy_v0.1.tflitezYhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/hey_rhasspy_v0.1.tflitez"resources/models/timer_v0.1.tflitezShttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/timer_v0.1.tflitez$resources/models/weather_v0.1.tflitezUhttps://github.com/dscripka/openWakeWord/releases/download/v0.5.1/weather_v0.1.tflite)ÚalexaÚhey_mycroftÚ
hey_jarvisÚhey_rhasspyÚtimerÚweatherr   Ú1_minute_timerÚ5_minute_timerÚ10_minute_timerÚ20_minute_timerÚ30_minute_timerÚ1_hour_timer)Ú1Ú2Ú3Ú4Ú5Ú6Útflitec                 C   s8   | dkrdd„ t  ¡ D ƒS | dkrdd„ t  ¡ D ƒS d S )Nr   c                 S   s   g | ]}t | d  ‘qS )r   )ÚMODELS©Ú.0Úi© r!   ú\/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/openwakeword/__init__.pyÚ
<listcomp>C   s    z.get_pretrained_model_paths.<locals>.<listcomp>Úonnxc                 S   s    g | ]}t | d   dd¡‘qS )r   z.tflitez.onnx)r   Úreplacer   r!   r!   r"   r#   E   s     )r   Úkeys)Úinference_frameworkr!   r!   r"   Úget_pretrained_model_pathsA   s
   ÿr(   )r   )ÚosÚopenwakeword.modelr   Úopenwakeword.vadr   Ú"openwakeword.custom_verifier_modelr   Ú__all__ÚpathÚjoinÚdirnameÚabspathÚ__file__ÚFEATURE_MODELSÚ
VAD_MODELSr   Úmodel_class_mappingsr(   r!   r!   r!   r"   Ú<module>   s\    þþûþÿ	þþþþþþëúÿ