import streamlit as st | |
from annotated_text import annotated_text | |
from utils import ner_extraction | |
st.title('NER with Fined-Tuned BERT model') | |
input_text = st.text_area("Please Enter the text..") | |
ner_extraction = ner_extraction(input_text) | |
if st.button('Submit'): | |
annotated_text(ner_extraction.entity_position()) | |