o
    *Îj
  ã                   @   s  d dl mZ d dlmZ d dlmZ dZdZdZej	eej	ej
ejejejgeej	iejeejejejejgeej	iejeejejejejgeej	iejeejejgeejiejeejejgeejiejeejgeejiejeejgeejiiZdd„ ZeeƒZd	efd
d„Zdd„ ZdS )é    )Ú
TaskModels)Úregistry)ÚTasksZ	sub_tasksZparent_taskZ
task_modelc                 C   s>   t ƒ }| D ]}| | t D ]}t|t| | t i||< qq|S ©N)ÚdictÚ	SUB_TASKSÚPARENT_TASKÚ
TASK_MODEL)Zforward_indexZinverted_indexÚindexÚitem© r   ú\/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/modelscope/utils/task_utils.pyÚ_inverted_indexG   s   
þÿr   Útaskc                 C   s
   | t jkS r   )r   Zsentence_embedding)r   r   r   r   Úis_embedding_taskU   s   
r   c                 C   s(   | t v rt |  t t |  t fS | d fS r   )ÚINVERTED_TASKS_LEVELr   r	   )Z	group_keyr   r   r   Úget_task_by_subtask_nameY   s   ÿ
ÿr   N)Zmodelscope.metainfor   Zmodelscope.utilsr   Zmodelscope.utils.constantr   r   r   r	   Ztext_classificationZsentence_similarityZsentiment_classificationZsentiment_analysisZnliZtoken_classificationZnamed_entity_recognitionZword_segmentationZpart_of_speechZtext_generationZtext2text_generationZinformation_extractionZrelation_extractionZ	fill_maskZtext_rankingZDEFAULT_TASKS_LEVELr   r   Ústrr   r   r   r   r   r   Ú<module>   sˆ    û÷üø
üø
þûþûÿüÿüÌ>