Ng} UddlmZddlZddlZddlmZmZmZmZddl m Z ddl m Z ddl mZddlm ZddlmZddlmZd Ziad ed <ddZe\ZZddZdddZGddeZdS)) annotationsN)DictOptionalTupleType) MODEL_TYPESUnstructuredDetectronONNXModel)UnstructuredModelUnstructuredYoloXModel)LazyDictyoloxzDict[str, UnstructuredModel]modelsreturnETuple[Dict[str, Type[UnstructuredModel]], Dict[str, dict | LazyDict]]cjidtDdtDittfS)zKdefault model mappings for models that are in `unstructured_inference` repoci|] }|t Sr .0names ^/var/www/html/ai-engine/env/lib/python3.11/site-packages/unstructured_inference/models/base.py z.get_default_model_mappings..s X X XD4/ X X Xci|] }|t Srr rs rrz.get_default_model_mappings..s F F FD4' F F Fr)DETECTRON2_ONNX_MODEL_TYPESYOLOX_MODEL_TYPESrrrget_default_model_mappingsrsP   X X.)sGGG$D+GGGrN)model_config_mapupdatemodel_class_map)r r"s `rregister_new_modelr(#sH L)))GGGG,GGGHHHHHr model_name Optional[str]c>|*tjd}||nt}|tvr t|Stjd}|et |5}t j|}d|dD}||d<dddn #1swxYwYn)|tvrt|}ntd|t|}|j di||t|<|S)z$Gets the model object by model name.NUNSTRUCTURED_DEFAULT_MODEL_NAME6UNSTRUCTURED_DEFAULT_MODEL_INITIALIZE_PARAMS_JSON_PATHc4i|]\}}t||Sr)int)rkeyvalues rrzget_model..>s1"""$.CC%"""r label_mapzUnknown model type: r) osenvironget DEFAULT_MODELropenjsonloaditemsr%UnknownModelExceptionr' initialize)r)default_name_from_envinitialize_param_jsonfpinitialize_paramslabel_map_int_keysmodels r get_modelrC,s " /P Q Q.C.O**Ub Vj!!JNN+cdd( ' ( ( @B $ " ""2CK2P2V2V2X2X""" .@ k *  @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ ) ) ) 0 <  '(Kz(K(KLL L.z:<B==CCceZdZdZdS)r;z6A model was requested with an unrecognized identifier.N)__name__ __module__ __qualname____doc__rrrr;r;Os@@Drr;)rr)r r!r"r )N)r)r*rr ) __future__rr8r3typingrrrr,unstructured_inference.models.detectron2onnxrrr /unstructured_inference.models.unstructuredmodelr #unstructured_inference.models.yoloxrr unstructured_inference.utilsrr6r__annotations__rr'r%r(rC Exceptionr;rrrrQsq""""""" ............XWWWWWMMMMMMPPPPPPFFFFFF111111 ')))))<<<<%?$>$@$@!!IIII     F     I     r