import spacy
import streamlit as st
import spacy_streamlit
from config import COLORS_NER,DEFAULT_CV

nlp = spacy.load("ner_model")

st.title("SpaCy NER Model")
st.header("Entidades en el texto de un curriculum-cv")

cv_text=st.text_input("Ingrese texto de cv aquí",DEFAULT_CV)

if cv_text is not None:
   doc= nlp(cv_text)
   spacy_streamlit.visualize_ner(doc,labels=nlp.get_pipe('ner').labels,colors = COLORS_NER)