saadul's picture
Update app.py
5b5117f
raw
history blame
656 Bytes
import streamlit as st
import spacy
# Load spaCy NLP model for NER
nlp = spacy.load("en_core_web_sm")
# Streamlit app
def main():
st.title("Named Entity Recognition (NER) Demo")
# User input
text_input = st.text_area("Enter text:", "John Doe is the CEO of ABC Corp, and it is located in New York.")
# NER processing
if st.button("Extract Entities"):
doc = nlp(text_input)
# Display entities
entities = [(ent.text, ent.label_) for ent in doc.ents]
st.write("Named Entities:")
for entity, label in entities:
st.write(f"- {entity} ({label})")
if __name__ == "__main__":
main()