o 0j @sddlmZmZmZmZmZddlZddlm Z ddl m Z m Z ddl mZddlmZd d lmZeje d Gd d d eZdS))AnyDictListOptionalUnionN)pipeline_requires_extra) HPIConfigPaddlePredictorOption) WhisperResult) benchmark) BasePipelineZspeechcseZdZdZdZddddddddedeedeed eeeefd ee d e d ee eeefe fd dffddZ de eeeejeejfd efddZZS)%MultilingualSpeechRecognitionPipelinez(Multilingual Speech Recognition PipelineZmultilingual_speech_recognitionNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc sDtjd||||||d||dd} || |_| ddS)a)Initializes the multilingual speech recognition pipeline. Args: config (Dict): Configuration dictionary containing model and other parameters. device (Optional[str], optional): The device to use for prediction. Defaults to `None`. engine (Optional[str], optional): Inference engine. Defaults to `None`. engine_config (Optional[Dict[str, Any]], optional): Engine-specific config. Defaults to `None`. pp_option (Optional[PaddlePredictorOption], optional): Paddle predictor options. Defaults to `None`. use_hpip (bool, optional): Whether to use HPIP. Defaults to `False`. hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional): HPIP configuration. Defaults to `None`. rZ SubModulesZMultilingualSpeechRecognitionZ batch_sizeN)super__init__Z create_model%multilingual_speech_recognition_model) selfrrrrrrrkwargsZ,multilingual_speech_recognition_model_config __class__r/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/multilingual_speech_recognition/pipeline.pyr!s"  z.MultilingualSpeechRecognitionPipeline.__init__inputcks||EdHdS)arPredicts speech recognition results for the given input. Args: input (Union[str, list[str], np.ndarray, list[np.ndarray]]): The input audio or path. **kwargs: Additional keyword arguments that can be passed to the function. Returns: WhisperResult: The predicted whisper results, support str and json output. N)r)rr#rrrr"predictMs z-MultilingualSpeechRecognitionPipeline.predict)__name__ __module__ __qualname____doc__entitiesrrstrrr boolrr rrnpZndarrayr r$ __classcell__rrr r"rs>  ,r)typingrrrrrnumpyr,Z utils.depsrmodelsr r Z-models.multilingual_speech_recognition.resultr Zutils.benchmarkr baserZ time_methodsrrrrr"s