Spaces:
Sleeping
Sleeping
from functions import * | |
# set the title | |
st.sidebar.title(DASHBOARD_TITLE) | |
# load the model | |
with st.spinner("Loading the model..."): | |
model = load_pipeline() | |
# input | |
text = st.text_area(value="Woman, 43, feeling pain in the stomach. No vomiting", label="Enter the text here") | |
submit = st.button("Submit") | |
if text and submit: | |
# get the entities | |
entities = model(text) | |
if len(entities) == 0: | |
st.error("No entities found") | |
# annotate the entities | |
annotated_entities = [annotate(entity) for entity in entities] | |
# display the entities | |
annotated_text(annotated_entities) | |