tjS+ddlZddlmZddlZddlZddlmZmZm Z m Z m Z ddl m Z mZerddlmZe jeZded ed ejfd Zeed Gdde ZdS)N)Any)add_end_docstringsis_torch_availableis_torchaudio_availableis_torchcodec_availablelogging)Pipelinebuild_pipeline_init_args),MODEL_FOR_AUDIO_CLASSIFICATION_MAPPING_NAMESbpayload sampling_ratereturnc ~|}d}d}dddd|d|d|d d d d g } tj|tjtj }n#t$rt dwxYw||}|d}t j|t j} | j ddkrt d| S)z? Helper function to read an audio file through ffmpeg. 1f32leffmpegz-izpipe:0z-acz-arz-fz -hide_bannerz -loglevelquietzpipe:1)stdinstdoutzFffmpeg was not found but is required to load audio files from filenamerzMalformed soundfile) subprocessPopenPIPEFileNotFoundError ValueError communicatenp frombufferfloat32shape) rraracformat_for_conversionffmpeg_commandffmpeg_process output_stream out_bytesaudios i/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/transformers/pipelines/audio_classification.py ffmpeg_readr+s  B B#      N c#). XbXghhh cccabbbc"..x88Ma I M)RZ 0 0E {1~./// Ls +AAT)has_feature_extractorc eZdZdZdZdZdZdZfdZde j e ze ze zdedee e efffd Zdd Zd Zd ZddZxZS)AudioClassificationPipelinea Audio classification pipeline using any `AutoModelForAudioClassification`. This pipeline predicts the class of a raw waveform or an audio file. In case of an audio file, ffmpeg should be installed to support multiple audio formats. Example: ```python >>> from transformers import pipeline >>> classifier = pipeline(model="superb/wav2vec2-base-superb-ks") >>> classifier("https://huggingface.co/datasets/Narsil/asr_dummy/resolve/main/1.flac") [{'score': 0.997, 'label': '_unknown_'}, {'score': 0.002, 'label': 'left'}, {'score': 0.0, 'label': 'yes'}, {'score': 0.0, 'label': 'down'}, {'score': 0.0, 'label': 'stop'}] ``` Learn more about the basics of using a pipeline in the [pipeline tutorial](../pipeline_tutorial) This pipeline can currently be loaded from [`pipeline`] using the following task identifier: `"audio-classification"`. See the list of available models on [huggingface.co/models](https://huggingface.co/models?filter=audio-classification). FTcd|vr|dd|d<n d|vrd|d<tj|i||tdS)Ntop_k)super__init__check_model_typer )selfargskwargs __class__s r*r3z$AudioClassificationPipeline.__init__csm f  !8"F7OO F " "F7O$)&))) JKKKKKinputsr7rc 8tj|fi|S)a Classify the sequence(s) given as inputs. See the [`AutomaticSpeechRecognitionPipeline`] documentation for more information. Args: inputs (`np.ndarray` or `bytes` or `str` or `dict`): The inputs is either : - `str` that is the filename of the audio file, the file will be read at the correct sampling rate to get the waveform using *ffmpeg*. This requires *ffmpeg* to be installed on the system. - `bytes` it is supposed to be the content of an audio file and is interpreted by *ffmpeg* in the same way. - (`np.ndarray` of shape (n, ) of type `np.float32` or `np.float64`) Raw audio at the correct sampling rate (no further check will be done) - `dict` form can be used to pass raw audio sampled at arbitrary `sampling_rate` and let this pipeline do the resampling. The dict must be either be in the format `{"sampling_rate": int, "raw": np.array}`, or `{"sampling_rate": int, "array": np.array}`, where the key `"raw"` or `"array"` is used to denote the raw audio waveform. top_k (`int`, *optional*, defaults to None): The number of top labels that will be returned by the pipeline. If the provided number is `None` or higher than the number of labels available in the model configuration, it will default to the number of labels. function_to_apply (`str`, *optional*, defaults to "softmax"): The function to apply to the model output. By default, the pipeline will apply the softmax function to the output of the model. Valid options: ["softmax", "sigmoid", "none"]. Note that passing Python's built-in `None` will default to "softmax", so you need to pass the string "none" to disable any post-processing. Return: A list of `dict` with the following keys: - **label** (`str`) -- The label predicted. - **score** (`float`) -- The corresponding probability. )r2__call__)r5r:r7r8s r*r<z$AudioClassificationPipeline.__call__ms%D uww11&111r9Nc i}||jjj|d<n+||jjjkr|jjj}||d<||dvrtd|d||d<nd|d<ii|fS)Nr0)softmaxsigmoidnonez'Invalid value for `function_to_apply`: z2. Valid options are ['softmax', 'sigmoid', 'none']function_to_applyr>)modelconfig num_labelsr)r5r0rAr7postprocess_paramss r*_sanitize_parametersz0AudioClassificationPipeline._sanitize_parameterss =*.**;*F w ' 'tz(333 )4*/ w '  ( (FFF G>OGGG7H 2 3 36? 2 32)))r9crt|tr|ds|drtj|dj}nct|dkr!|jdd}n3|dkr |jd}n |jd}||\}}|}|}fdt ||D}|S)Nr>rr?cJg|]\}}|jjj|d S))scorelabel)rBrCid2label).0r_idr5s r* z;AudioClassificationPipeline.postprocess..s5pppQ[QVX[EDJ,=,Fs,KLLpppr9)logitsr>r?topktolistzip)r5r{r0rAprobsscoresidslabelss` r* postprocessz'AudioClassificationPipeline.postprocesss  ) )!(+33B77EE ) + +!(+3355EE!(+Ejj'' jjllpppp_bcikn_o_oppp r9)NN)r1r>)__name__ __module__ __qualname____doc___load_processor_load_image_processor_load_feature_extractor_load_tokenizerr3rrkrYrRrerlistr<rFrwr|r __classcell__)r8s@r*r.r.Bs2O!"OLLLLL"2rzE1C7$>"2#"2RVW[\_ad\dWeRf"2"2"2"2"2"2H****,IIIVr9r.)rtypingrrTr^rutilsrrrrr baser r models.auto.modeling_autor get_loggerrloggerrYintrkr+r.ryr9r*rsF uuuuuuuuuuuuuu44444444YXXXXXX  H % %!%!! !!!!H,,4HHHIIAAAAA(AAJIAAAr9