Path: blob/main/fastapi/__pycache__/working.cpython-38.pyc
744 views
U
X3{b � @ sl d dl Z d dlmZ d dlmZ d dlmZ e� ZdZe � e�Z
G dd� de�Z e� d�e d �d
d ��Z
dS ) � N)�FastAPI)� BaseModel)�Listz ../models/model_v3.0/model-bestc @ s$ e Zd ZU eed< G dd� d�ZdS )�Input�textc @ s e Zd ZdddiiZdS )z Input.ConfigZexampler z@The Urban Redevelopment Authority is located at 45 Maxwell Road.N)�__name__�
__module__� __qualname__Z schema_extra� r
r
� .\working.py�Config s
��r N)r r r �str�__annotations__r r
r
r
r r s
r z/find-locations-POST)�inputc C sD t | j�}g }|jD ]&}i }|j|d< |j|d< |�|� qd|iS )NZentity�label�entities)�nlpr ZentsZlabel_�append)r �docZlist_of_entitiesZentZ
ent_resultr
r
r �find_locations_post s
r )Zspacy�fastapir Zpydanticr �typingr �appZ
model_path�loadr r Zpostr r
r
r
r �<module> s