
    Pi݀                         d Z ddlmZ ddlZddlmZ ddlmZ dede	e
ef         dz  fdZdede	e
ef         dz  fd	Zdede	e
ef         fd
ZdS )z
Defines internal helper methods for handling transformers and diffusers pipelines.
These are used by load_from_pipeline method in pipelines.py.
    )AnyN)Image)
componentspipelinereturnc                 	   	 dd l n"# t          $ r}t          d          |d }~ww xY wdt          ffd} || d          r2t          j        ddd	          t          j        d
d          d d dS  || d          r2t          j        ddd	          t          j        dd          d d dS  || d          r1t          j        dd          t          j        dd          d d dS  || d          r1t          j        dd          t          j        dd          d d dS  || d          r2t          j        ddd	          t          j        dd          d d dS  || d          r^t          j        dd d!          t          j        d"d          gt          j        d#d          t          j        d$d          gd% d& dS  || d'          r2t          j        ddd!          t          j        d(d          d) d* dS  || d+          r1t          j        dd          t          j        dd          d, d- dS  || d.          r1t          j        dd          t          j	        d/d          d0 d1 dS  || d2          r1t          j        dd          t          j        dd          d3 d4 dS  || d5          r1t          j        dd          t          j        d6d          d7 d8 dS  || d9          r1t          j        dd          t          j        d:d          d; d< dS  || d=          r\t          j        dd          t          j        d>d          t          j
        d?d          gt          j        dd          d@ dA dS  || dB          rHt          j        ddCd	          t          j        d"d          gt          j        dDd          dE dF dS  || dG          rHt          j        ddd	          t          j        d"d          gt          j        d$d          dH dI dS  || dJ          r2t          j        ddd	          t          j        dKd          dL dM dS  || dN          r2t          j        ddd	          t          j        dOd          dP dQ dS t          dRt          |                      )SNr   zAtransformers not installed. Please try `pip install transformers`
class_namec                 J    t          |d           }|ot          | |          S Ngetattr
isinstance)r   r	   clstransformerss      j/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/gradio/pipelines_utils.pyis_transformers_pipeline_typezChandle_transformers_pipeline.<locals>.is_transformers_pipeline_type   s)    lJ550z(C000    AudioClassificationPipelinefilepathInputFtypelabelrenderClassr   r   c                 
    d| iS Ninputs is    r   <lambda>z.handle_transformers_pipeline.<locals>.<lambda>    
    XqM r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r   scorer    .0r"   s     r   
<dictcomp>zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>!   "    %H%H%Haj!G*%H%H%Hr   r    rs    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>!       %H%Ha%H%H%H r   r   outputs
preprocesspostprocess"AutomaticSpeechRecognitionPipelineOutputc                 
    d| iS r   r    r!   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>'   r$   r   c                     | d         S Ntextr    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>(   
    QvY r   FeatureExtractionPipelinec                 
    d| iS r   r    xs    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>.   r$   r   c                     | d         S Nr   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>/   s
    QqT r   FillMaskPipelineClassificationc                 
    d| iS r   r    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>5   r$   r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S 	token_strr(   r    r)   s     r   r+   zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>6   "    %L%L%LQanaj%L%L%Lr   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>6       %L%L!%L%L%L r   ImageClassificationPipelineInput Imagec                 
    d| iS Nimagesr    r!   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>>   r$   r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r)   s     r   r+   zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>?   r,   r   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>?   r/   r   QuestionAnsweringPipeline   Context)linesr   r   QuestionAnswerScorec                     | |dS )N)contextquestionr    cqs     r   r#   z.handle_transformers_pipeline.<locals>.<lambda>K   s    1!'D'D r   c                 "    | d         | d         fS Nanswerr(   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>L       ak1W:%> r   SummarizationPipelineSummaryc                 
    d| iS r   r    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>R   r$   r   c                     | d         d         S Nr   summary_textr    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>S       QqT.%9 r   TextClassificationPipelinec                     | gS r   r    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>Y       QC r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r)   s     r   r+   zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>Z   r,   r   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>Z   r/   r   TokenClassificationPipelineEntitiesc                     | gS r   r    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>`   rk   r   c                     || dS )N)r9   entitiesr    )r.   r9   s     r   r#   z.handle_transformers_pipeline.<locals>.<lambda>a   s    , , r   TextGenerationPipelinec                 
    d| iS )Ntext_inputsr    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>j   s    ]A$6 r   c                     | d         d         S Nr   generated_textr    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>k       QqT*:%; r   TranslationPipelineTranslationc                     | gS r   r    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>q   rk   r   c                     | d         d         S Nr   translation_textr    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>r       QqT*<%= r   Text2TextGenerationPipelineGenerated Textc                     | gS r   r    r=   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>x   rk   r   c                     | d         d         S rw   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>y   ry   r   ZeroShotClassificationPipeline&Possible class names (comma-separated)zAllow multiple true classesc                     | ||dS )N)	sequencescandidate_labelsmulti_labelr    )r"   r\   ms      r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   s    $% + + r   c                 `      fdt          t           d                             D             S )Nc                 F    i | ]}d          |         d         |         S )labelsscoresr    )r*   r"   r.   s     r   r+   zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>   s8     & & &34(A(A& & &r   r   )rangelenr-   s   `r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   s?     & & & &8=c!H+>N>N8O8O& & & r   !DocumentQuestionAnsweringPipelineInput DocumentLabelc                     | |dS N)imagerZ   r    imgr]   s     r   r#   z.handle_transformers_pipeline.<locals>.<lambda>       3A)F)F r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r`   r(   r    r)   s     r   r+   zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>   "    %I%I%I!ak1W:%I%I%Ir   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>       %I%Iq%I%I%I r   VisualQuestionAnsweringPipelinec                     | |dS r   r    r   s     r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   r   r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r   r    r)   s     r   r+   zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<dictcomp>   r   r   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   r   r   ImageToTextPipelineTextc                 
    d| iS rM   r    r!   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   r$   r   c                     | d         d         S rw   r    r-   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   ry   r   ObjectDetectionPipelineObjects Detectedc                 
    d| iS r   r    r!   s    r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   r$   r   c                     |d | D             fS )Nc                     g | ]@}|d          d         |d          d         |d          d         |d          d         f|d         fAS )boxxminyminxmaxymaxr   r    r)   s     r   
<listcomp>zBhandle_transformers_pipeline.<locals>.<lambda>.<locals>.<listcomp>   sj         eHV,eHV,eHV,eHV,	 '
  r   r    )r.   r   s     r   r#   z.handle_transformers_pipeline.<locals>.<lambda>   s*        + r   z(Unsupported transformers pipeline type: )r   ImportErrorstrr   Audior   Textbox	Dataframer   HighlightedTextCheckboxAnnotatedImage
ValueErrorr   )r   ier   r   s      @r   handle_transformers_pipeliner      sl      O
 
	
1C 1 1 1 1 1 1 %$X/LMM 
 &JgeTTT!'geDDD11HH	
 
 	
 %$X/STT 
 &JgeTTT!)GGG11..	
 
 	
 %$X/JKK 
 (wuEEE!+(5III11)>	
 
 	
 %$X/ABB 
 (wuEEE!'.>uMMM11LL	
 
 	
 %$X/LMM 
 &}U   "'.>uMMM11HH
 
 	
 %$X/JKK 
 ")EJJJ"EBBB
 "%@@@ wu=== ED>>
 
 	
 %$X/FGG 
 (qNNN!)	%HHH1199	
 
 	
 %$X/KLL 
 (wuEEE!'.>uMMM'-HH	
 
 	
 %$X/LMM 	
 (wuEEE!1
5QQQ'- 	
 
 	
 %$X/GHH 
 (wuEEE!)GGG66;;	
 
 	
 %$X/DEE 
 (wuEEE!)eLLL'-==	
 
 	
 %$X/LMM 
 (wuEEE!)0@OOO'-;;	
 
 	
 %$X/OPP 
 "???"B5   #*GPUVVV "'.>uMMM 
 
 
 	
$ %$X/RSS 	
  j8HQVWWW"EBBB "'geDDDFFII
 
 	
 %$X/PQQ 	
  jeTTT"EBBB "'geDDDFFII
 
 	
 %$X/DEE 
 &}U   ")uEEE11;;
 
 	
 %$X/HII 
 &}U   "0(   21 
 
 	
0 PXPP
Q
QQ    
'"'c                 	   	 dd l n"# t          $ r}t          d          |d }~ww xY wdt          ffd} || d          rxt          j        dd          t          j        d	d          t          j        d
dddd          t          j        ddddd          gt          j        ddd          d d dS  || d          rt          j        dd          t          j        d	d          t          j        ddd          t          j        ddddd          t          j        d
dddd          t          j        ddddd          gt          j        ddd          d  d! dS  || d"          rt          j        dd          t          j        d	d          t          j        ddd          t          j        dd#d          t          j        ddddd          t          j        d
dddd          t          j        ddddd          gt          j        ddd          d$ d% dS  || d&          rt          j        dd          t          j        d	d          t          j        ddd          t          j        ddddd          t          j        d
dddd          t          j        ddddd          gt          j        ddd          d' d( dS  || d)          rdt          j        ddd          t          j        d
dddd          t          j        ddddd          gt          j        ddd          d* d+ dS  || d,          rt          j        dd          t          j        d	d          t          j        ddd          t          j        d
dddd          t          j        ddddd          t          j        d-dd.d/d          gt          j        ddd          d0 d1 dS  || d2          rt          j        dd          t          j        d	d          t          j        ddd          t          j        d
dddd          t          j        ddddd          t          j        d3dd4dd          gt          j        ddd          d5 d6 dS t          d7t          |                      )8Nr   z;diffusers not installed. Please try `pip install diffusers`r	   c                 J    t          |d           }|ot          | |          S r   r   )r   r	   r   	diffuserss      r   is_diffusers_pipeline_typez=handle_diffusers_pipeline.<locals>.is_diffusers_pipeline_type   s)    iT220z(C000r   StableDiffusionPipelinePromptFr   zNegative promptzNumber of inference steps     2   r   minimummaximumvaluestepzGuidance scale   g      @g      ?zGenerated Imagepil)r   r   r   c                     | |||dS )N)promptnegative_promptnum_inference_stepsguidance_scaler    )r   n_promptnum_inf_stepsg_scales       r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>   s     #+'4")	L L r   c                     | d         d         S NrN   r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>       Qx[^ r   r0   StableDiffusionImg2ImgPipeliner   r   r   Strengthg?g?c                 ^    | t          j        |                              d          ||||dS N   r   )r   r   r   r   r   strengthr   openresizer   r   r   r   r   r   s         r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  =     !E**11*==#+'4")$  r   c                     | d         d         S r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  r   r   StableDiffusionInpaintPipelinez
Mask Imagec                     | t          j        |                              d          t          j        |                              d          ||||dS )Nr   )r   r   
mask_imager   r   r   r   r   )r   r   r   r   r   r   r   s          r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>8  sX     !E**11*==#j44;;JGG#+'4")$  r   c                     | d         d         S r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>G  r   r    StableDiffusionDepth2ImgPipelinec                 ^    | t          j        |                              d          ||||dS r   r   r   s         r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>d  r   r   c                     | d         d         S r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>q  r   r   %StableDiffusionImageVariationPipelinec                 X    t          j        |                               d          ||dS )Nr   )r   r   r   r   )r   r   r   s      r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  s1    E**11*=='4")A A r   c                     | d         d         S r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  r   r   &StableDiffusionInstructPix2PixPipelinezImage Guidance scale   g      ?c                 ^    | t          j        |                              d          ||||dS )Nr   )r   r   r   r   r   image_guidance_scaler   )r   r   r   r   r   img_g_scales         r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  s=     !E**11*==#+'4")(3  r   c                     | d         d         S r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  r   r   StableDiffusionUpscalePipelinezNoise leveld   c                 ^    | t          j        |                              d          ||||dS )Nr   )r   r   r   r   r   noise_levelr   )r   r   r   r   r   r   s         r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  s=     !E**11*==#+'4")*  r   c                     | d         d         S r   r    r-   s    r   r#   z+handle_diffusers_pipeline.<locals>.<lambda>  r   r   z%Unsupported diffusers pipeline type: )	r   r   r   r   r   Sliderr   r   r   )r   r   r   r   s      @r   handle_diffusers_pipeliner      s`      I
 
	
1 1 1 1 1 1 1 "!(,EFF 
 "%@@@"):5III!5   !*  $ "''E    439
 
 	
< "!(,LMM )
 "%@@@"):5III jNNN!$a#C   !5   !*  , "''E    43O(
 (
 (	
R "!(,LMM ,
 "%@@@"):5III jNNN jUSSS!$a#C   !5   !*  . "''E    43U+
 +
 +	
X "!(,NOO )
 "%@@@"):5III jNNN!$a#C   !5   !*  , "''E    43O(
 (
 (	
R "!(,STT 
  jNNN!5   !*  " "''E   
 435
 
 	
8 "!(,TUU -
 "%@@@"):5III jNNN!5   !*   !0  %4 "''E    43W,
 ,
 ,	
Z "!(,LMM )
 "%@@@"):5III jNNN!5   !*   !'CrPQ  %, "''E    43O(
 (
 (	
R MT(^^MM
N
NNr   c                     	 ddl mm n"# t          $ r}t          d          |d }~ww xY w j        dk    r.t          j        d          t          j        d          d d d	S  j        d
k    rZt          j        dd          t          j        d          gt          j        d          t          j        d          gd d d	S  j        dk    rIt          j        dd          t          j        ddddd          gt          j        d          d d d	S  j        dk    rEt          j        d          t          j	        dd          gt          j        d          d  d! d	S  j        d"k    r.t          j        d          t          j        d#          d d$ d	S  j        d%k    rHt          j        d          t          j        ddddd          gt          j        d&          d' d( d	S  j        d)k    r.t          j        d          t          j
        d#          d d d*d+S  j        d,v rXt          j        d          t          j        d-          t          j        d.          gt          j        d/          d0 d1 d	S  j        d2k    rDt          j        d          t          j        d3          gt          j        d          d4 d5 d	S  j        d6k    r.t          j        d          t          j        d#          d d7 d	S  j        d8k    r2t          j        d9d:;          t          j        d<          fd=d> d	S  j        d?k    rHt          j        d9d:;          t          j	        dd          gt          j        d          fd@dA d	S  j        dBk    r3t          j        d9d:;          t          j        dC          fdDdE d*d+S  j        dFk    r2t          j        d9d:;          t          j        dG          fdHdI d	S  j        dJk    r3t          j        d9d:;          t          j        dK          fdLdM d*d+S  j        dNk    r2t          j        d9d:;          t          j        d#          fdOdP d	S  j        dQk    r3t          j        d9d;          t          j        dR           fdSdT d	S  j        dUk    r3t          j        d9d;          t          j        d#           fdVdW d	S  j        dXk    rDt          j        d          t          j        dY          gt          j        d#          dZ d[ d	S  j        d\k    rGt          j        d9d];          t          j        d          gt          j        d^          fd_d` d	S  j        dak    r2t          j        d9d:;          t          j        d#          fdbdc d	S  j        ddk    rHt          j        d9d;          t          j        d3          gt          j        d           fdedf d	S  j        dgk    rGt          j        d9d:;          t          j        d3          gt          j        d          fdhdi d	S  j        djk    rHt          j        d9d:;          t          j        d3          gt          j        dK          fdkdl d*d+S t          dm j                   )nNr   )as_url
read_audiozmtransformers_js_py not installed. Please add `transformers_js_py` to the requirements of your Gradio-Lite appz	fill-maskr   )r   rB   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S rF   r    r)   s     r   r+   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<dictcomp>  rH   r   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  rI   r   r0   zquestion-answeringrR   rS   )rT   r   rU   rV   rW   c                 
    || fS r   r    r[   s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    ( r   c                 "    | d         | d         fS r_   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>	  ra   r   summarizationz)The maximum numbers of tokens to generater   r   r   r   rc   c                     | d|ifS Nmax_new_tokensr    r9   r  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>      !>28 r   c                     | d         d         S rf   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  rh   r   ztext-classificationzTop kr   )r   r   c                     | d|ifS Ntopkr    )r9   r  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>%  s    dVTN-C r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r)   s     r   r+   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<dictcomp>&  r,   r   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>&  r/   r   ztext-generationr5   c                     | d         d         S rw   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>-  ry   r   ztext2text-generationr   c                     | d|ifS r  r    r	  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda><  r
  r   c                     | d         d         S rw   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>@  ry   r   ztoken-classificationT)r   r1   r2   r3   postprocess_takes_inputs>   translationtranslation_xx_to_yyzSource LanguagezTarget Languager{   c                     | ||dfS )N)src_langtgt_langr    )r>   sts      r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>R  s    11!.L.L*M r   c                     | d         d         S r~   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>S  r   r   zzero-shot-classificationr   c                 D    | d |                     d          D             fS )Nc                 6    g | ]}|                                 S r    stripr*   r\   s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>^       :::q:::r   ,split)r9   
classnamess     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>\  s+    ::J$4$4S$9$9:::4 r   c                 X    t          t          | d         | d         d                    S )Nr   r   F)strict)dictzipresults    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>`  s,    $F8$fX&6uEEE+ + r   zfeature-extractionc                 6    |                                  d         S r@   to_numpytensors    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>i  s    &//*;*;A*> r   zdepth-estimationr   rK   )r   r   Depthc                      |           fS r   r    
image_pathr   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>q      ffZ.@.@-B r   c                 6    | d                                          S )Ndepthto_pilr+  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>r  s    &/*@*@*B*B r   zimage-classificationc                 "     |           d|ifS r  r    )r5  r  r   s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>{  s    FF:4F4FQU3W r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r*   items     r   r+   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<dictcomp>|  s/     + + +15WtG}+ + +r   r    r+  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>|  s      + +9?+ + + r   zimage-segmentationSegmentationc                      |           fS r   r    r4  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r6  r   c                     |d | D             fS )Nc                     g | ]@}|d                                           dddddf         dz  |d          d|d          dfAS )maskNr   g     o@r    (r(   )r.  r>  s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>  sv        	 V--//111a8 =<<DM<<<  r   r    r,  r5  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s*      !'  
7 r   zimage-to-imagezOutput Imagec                      |           fS r   r    r4  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r6  r   c                 *    |                                  S r   r9  r+  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    &--// r   zobject-detectionr   c                      |           fS r   r    r4  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r6  r   c                     |d | D             fS )Nc           	      
   g | ]}t          |d          d                   t          |d          d                   t          |d          d                   t          |d          d                   f|d          d|d          dfS 	r   r   r   r   r   r   rE  r(   rF  intr>  s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>            UF 344UF 344UF 344UF 344	  =<<DM<<<  r   r    rG  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s*      !'  7 r   zimage-feature-extractionc                      |           fS r   r    r4  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r6  r   c                 *    |                                  S r   r.  r0  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    &//*;*; r   zaudio-classificationr   c                 H     | j         j        j        d                   fS Nsampling_rate	processorfeature_extractorconfigr"   r   r   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  ,    
x);B?S % r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r)   s     r   r+   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<dictcomp>  r,   r   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r/   r   zautomatic-speech-recognitionc                 H     | j         j        j        d                   fS rT  rV  rZ  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r[  r   c                     | d         S r8   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r:   r   ztext-to-audiozSpeaker Embeddingsc                     | d|ifS )Nspeaker_embeddingsr    )r9   ra  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    %'9:< r   c                 F    | d         t          j        | d                   fS )NrU  audio)npasarrayr-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    a&8"*QwZ:P:P%Q r   zdocument-question-answeringr   r   c                      |           |fS r   r    )r   r]   r   s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    &&++q)9 r   c                     | d         d         S )Nr   r`   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s    QqT& r   zimage-to-textc                      |           fS r   r    r4  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  r6  r   c                     | d         d         S rw   r    r-   s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  ry   r   zzero-shot-audio-classificationc                      | j         j        j        d                   d |                    d          D             fS )NrU  c                 6    g | ]}|                                 S r    r  r!  s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>  r"  r   r#  )rW  rX  rY  r%  )
audio_pathr&  r   r   s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  sN    
&8?P  ;:J$4$4S$9$9:::: r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r)   s     r   r+   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<dictcomp>  "    *R*R*Ra1W:qz*R*R*Rr   r    r+  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>      *R*R6*R*R*R r   zzero-shot-image-classificationc                 X     |           d |                     d          D             fS )Nc                 6    g | ]}|                                 S r    r  r!  s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>	  r"  r   r#  r$  r5  r&  r   s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  6    z""::J$4$4S$9$9:::: r   c                     d | D             S )Nc                 ,    i | ]}|d          |d         S r'   r    r)   s     r   r+   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<dictcomp>  ro  r   r    r+  s    r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  rp  r   zzero-shot-object-detectionc                 X     |           d |                     d          D             fS )Nc                 6    g | ]}|                                 S r    r  r!  s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>  r"  r   r#  r$  rs  s     r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  rt  r   c                     |d | D             fS )Nc           	      
   g | ]}t          |d          d                   t          |d          d                   t          |d          d                   t          |d          d                   f|d          d|d          dfS rM  rN  r>  s     r   r   zEhandle_transformers_js_pipeline.<locals>.<lambda>.<locals>.<listcomp>  rP  r   r    )r,  r5  _s      r   r#   z1handle_transformers_js_pipeline.<locals>.<lambda>  s*      !'  : r   z.Unsupported transformers_js_py pipeline type: )transformers_js_pyr   r   r   taskr   r   r   r   NumberJSONr   r   r   r   r   )r   r   r   r   s   ` @@r   handle_transformers_js_pipeliner    s*
   999999999   {
 
	 }## (w777!'.>???LL	
 
 	
 },,, ")<<<"444
 "222 w///  ?>
 
 	
 }'' "':::!E  	 ")	:::  :9!
 
 	
$ }--- "111!q999 "'.>???CCHH
 
 	
 }))) (w777!)999;;	
 
 	
 }... "111!E  	 ")0@AAA  <;!
 
 	
$ }... (w777!X666(,
 
 	
 }??? "111"):;;;"):;;;
 ")>>>MM==	
 	
 		
 }222 "111")QRRR "'.>???  
 
 	
 },,, (w777!+(;;;>>	
 
 	
 }*** &JmLLL!'g666BBBBBB	
 
 	
 }...  jFFF!q999 "'.>???WWWW 

 

 
	
 },,, &JmLLL!0~FFFBBBB
 
 )-
 
 	
" }((( &JmLLL!'n===BBBB99	
 
 	
 }*** &JmLLL!07IJJJBBBB  )-'
 
 	
* }222 &JmLLL!+(;;;BBBB;;	
 
 	
 }... &JgFFF!'g666    
 IH	
 	
 		
 }666 &JgFFF!)999    
 /.	
 	
 		
 }'' "111")=>>> "'h777  RQ
 
 	
 }555  j8HIII"444 ")8889999 

 

 
	
 }'' &JmLLL!)999BBBB;;	
 
 	
 }888  j@@@")QRRR "'.>???     SR
 
 	
 }888  jFFF")QRRR "'.>???    SR
 
 	
 }444  jFFF")QRRR "07IJJJ     )-3
 
 	
8 UhmUU
V
VVs    
-(-)__doc__typingr   numpyrd  PILr   gradior   r)  r   r   r   r  r    r   r   <module>r     s   
                      vR3 vR4S>D3H vR vR vR vRraO aOS#X0E aO aO aO aOH	Wc Wd38n W W W W W Wr   