{-j ddlmZmZmZmZddlmZddlmZm Z ddl m Z ddl m Z dd lmZe jed Gd d eZd S))AnyDictOptionalUnion)pipeline_requires_extra) HPIConfigPaddlePredictorOption) DocVLMResult) benchmark) BasePipeline multimodalceZdZdZdZddddddddedeedeed eeeefd ee d e d ee eeefe fd dffdZ ded efdZdZxZS)DocUnderstandingPipelinezDoc Understanding Pipelinedoc_understandingNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc tjd||||||d||didddi} || |_dS)aInitializes the doc understanding 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`. r SubModulesDocUnderstandingmodel_config_errorz)config error for doc_understanding_model!N)super__init__get create_modeldoc_understanding_model) selfrrrrrrrkwargsdoc_understanding_model_config __class__s v/var/www/html/banglarbhumi/venv/lib/python3.11/site-packages/paddlex/inference/pipelines/doc_understanding/pipeline.pyr#z!DocUnderstandingPipeline.__init__s2  '!      *0L")E)E)I)I  !#N O* * &(,'8'89W'X'X$$$inputc+2K|j|fi|Ed{VdS)a0Predicts doc understanding results for the given input. Args: input (dict): The input image and query. **kwargs: Additional keyword arguments that can be passed to the function. Returns: DocVLMResult: The predicted doc understanding results. N)r&)r'r-r(s r+predictz DocUnderstandingPipeline.predictHs;04/@@@@@@@@@@@@@r,c8|jdS)N)r&close)r's r+r1zDocUnderstandingPipeline.closeTs $**,,,,,r,)__name__ __module__ __qualname____doc__entitiesrrstrrr boolrr r#r r/r1 __classcell__)r*s@r+rrs0%$"H !% $2659AE'Y'Y'Y'Y 'Y  'Y  S#X/ 'Y12'Y'YU4S>9#<=>'Y 'Y'Y'Y'Y'Y'YR AT A A A A A-------r,rN)typingrrrr utils.depsrmodelsr r models.doc_vlm.resultr utils.benchmarkr baser time_methodsrr!r,r+rAs.-----------22222266666666111111(((((( &&;-;-;-;-;-|;-;-'&;-;-;-r,