o
    Tg                     @   sr   d dl Z de jd< d dlmZmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ d d	lmZ dd
efddZdS )    N1PYTORCH_ENABLE_MPS_FALLBACK)DetectionPredictorInlineDetectionPredictor)LayoutPredictor)OCRErrorPredictor)RecognitionPredictor)TableRecPredictor)TexifyPredictorreturnc              
   C   sL   t | |dt| |dt| |dt| |dt| |dt| |dt| |ddS )Ndevicedtype)layout_modeltexify_modelrecognition_modeltable_rec_modeldetection_modelinline_detection_modelocr_error_model)r   r
   r   r	   r   r   r   r    r   L/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/marker/marker/models.pycreate_model_dict   s   






r   )NN)osenvironsurya.detectionr   r   surya.layoutr   surya.ocr_errorr   surya.recognitionr   surya.table_recr	   surya.texifyr
   dictr   r   r   r   r   <module>   s    
