o
     ~ÖiÈ  ã                   @   sz   d dgZ ddlZddlZddlZej dej ej e¡d¡¡ ddl	m
Z
mZmZmZmZmZmZ dd„ Zdd
d„ZdS )ÚtorchÚ
torchaudioé    NÚsrc)Úinit_jit_modelÚget_speech_timestampsÚ
save_audioÚ
read_audioÚVADIteratorÚcollect_chunksÚOnnxWrapperc                 C   sP   |   d¡d   d¡}g }|D ]}z	| t|ƒ¡ W q   | d¡ Y qt|ƒS )Nú+r   Ú.)ÚsplitÚappendÚintÚtuple)ÚvÚsplittedÚversion_listÚi© r   úC/home/ubuntu/.cache/torch/hub/snakers4_silero-vad_master/hubconf.pyÚversiontuple   s   r   Fé   c           
      C   sÌ   ddg}| r||vrt d|› ƒ‚| s+tj}d}t|ƒt|ƒk r+t d|› d|› dƒ‚tj tj t¡dd	d
¡}| rR|dkrAd}nd|› d}t	tj ||¡|ƒ}n	t
tj |d¡ƒ}tttttf}	||	fS )z”Silero Voice Activity Detector
    Returns a model with a set of utils
    Please see https://github.com/snakers4/silero-vad for usage examples
    é   r   zAvailable ONNX opset_version: z1.12.0zPlease install torch z or greater (z installed)r   Ú
silero_vadÚdatazsilero_vad.onnxÚsilero_vad_16k_opz.onnxzsilero_vad.jit)Ú	Exceptionr   Ú__version__r   ÚosÚpathÚjoinÚdirnameÚ__file__r   r   r   r   r   r	   r
   )
ÚonnxÚforce_onnx_cpuÚopset_versionÚavailable_opsÚinstalled_versionÚsupported_versionÚ	model_dirÚ
model_nameÚmodelÚutilsr   r   r   r      s,   ür   )FFr   )Údependenciesr   r    Úsysr!   Úinsertr"   r#   r$   Úsilero_vad.utils_vadr   r   r   r   r	   r
   r   r   r   r   r   r   r   Ú<module>   s     $	