o
    q¦µi¬  ã                   @   s0   d Z ddlmZ ddlmZ G dd„ deƒZdS )z#Parser for performing ner detectioné    )Úpredict_neré   )ÚParserc                   @   s    e Zd ZdZdd„ Zdd„ ZdS )Ú	NerParserz'This parser add pos tags for Urdu text.c                 C   s   dS )ÚpassN© )ÚselfÚconfigr   r   úQ/home/ubuntu/.local/lib/python3.10/site-packages/urduhack/pipeline/parsers/ner.pyÚ_set_up   s    zNerParser._set_upc                 C   sh   |j D ].}t|jƒ}t|ƒt|jƒksJ dƒ‚t||jƒD ]\}}|d |jkr0t|d|d ƒ qq|S )z%Function to normalize|preprocess textz Error in Ner tagsr   Úneré   )Ú	sentencesr   ÚtextÚlenÚtokensÚzipÚsetattr)r   ÚdocumentÚsentenceÚtagsÚtagÚtokenr   r   r
   Úparse   s   

€þzNerParser.parseN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r
   r   	   s    r   N)r   Úurduhack.models.nerr   Úparserr   r   r   r   r   r
   Ú<module>   s   