
    &i                        d dl mZ d dlZd dlZd dlZd dlmZ ej                            dd           ej                            dd           d dl	m
Z
 ddZedk    r e e                      dS )    )annotationsN)PathKMP_DUPLICATE_LIB_OKTRUETOKENIZERS_PARALLELISMfalse)WhisperModelreturnintc                    t          j                    } |                     d           |                     d           |                     dd           |                     dd           |                                 }t	          |j        dd	
          }|                    |j        |j        dddd          \  }}d |D             }t          |j
                                      t          j        |d          d           dS )Ninputoutput_jsonz--modelzsmall.en)defaultz
--languageencpuint8)devicecompute_type   TF)language	beam_size
vad_filterword_timestampscondition_on_previous_textc                    g | ]c}|j         	|j                                         "t          |j                  t          |j                  |j                                         d dS ))startendtext)r   stripfloatr   r   ).0segs     scripts/_fw_transcribe.py
<listcomp>zmain.<locals>.<listcomp>    sv        8 ((39%%>>HNN$$	
 	
         )indentzutf-8)encodingr   )argparseArgumentParseradd_argument
parse_argsr	   model
transcriber   r   r   r   
write_textjsondumps)parserargsr-   segments_payloads         r#   mainr7      s   $&&F
   
&&&
	:666
d333DEGGGE""
#' #  KHa    G 		%%dj&C&C&Cg%VVV1r%   __main__)r
   r   )
__future__r   r)   r0   ospathlibr   environ
setdefaultfaster_whisperr	   r7   __name__
SystemExit r%   r#   <module>rB      s    " " " " " "   				       
  ,f 5 5 5 
  . 8 8 8 ' ' ' ' ' '   < z
*TTVV

 r%   