Update src/streamlit_app.py

src/streamlit_app.py

@@ -77,7 +77,7 @@ st.set_page_config(layout="wide", page_title="Named Entity Recognition App")
 st.subheader("StoryCraft", divider="blue")
 st.link_button("by nlpblogs", "https://nlpblogs.com", type="tertiary")
 expander = st.expander("**Important notes**")
-expander.write("""**Named Entities:** This StoryCraft predicts eighteen (18) labels: "Person", "Organization", "Location", "Date", "Time", "Quantity", "Product", "Event", "Title", "Job_title", "Artwork", "Media", "URL", "Website", "Hashtag", "Email_address", "IP_address", "File_path"
+expander.write("""**Named Entities:** This StoryCraft predicts eighteen (18) labels: "Person", "Organization", "Location", "Date", "Time", "Quantity", "Product", "Event", "Title", "Job_title", "Artwork", "Media", "URL", "Website", "Hashtag", "Email", "IP_address", "File_path"
 
 Results are presented in easy-to-read tables, visualized in an interactive tree map, pie chart and bar chart, and are available for download along with a Glossary of tags.      
 
@@ -119,7 +119,7 @@ if not comet_initialized:
     st.warning("Comet ML not initialized. Check environment variables.")
 
 # --- Label Definitions ---
-labels = ["Person","Organization","Location","Date","Time","Quantity","Product","Event","Title","Job_title","Artwork","Media","URL","Website","Hashtag","Email_address","IP_address","File_path"]
+labels = ["Person","Organization","Location","Date","Time","Quantity","Product","Event","Title","Job_title","Artwork","Media","URL","Website","Hashtag","Email","IP_address","File_path"]
 
 # Corrected mapping dictionary
 
@@ -127,7 +127,7 @@ labels = ["Person","Organization","Location","Date","Time","Quantity","Product",
 category_mapping = {
     "Core Foundational Entities": ["Person", "Organization", "Location", "Date", "Time", "Quantity"],
     "Content Enrichment Entities": ["Product", "Event", "Title", "Job_title", "Artwork", "Media"],
-    "Digital & Technical Entities": ["URL", "Website", "Hashtag", "Email_address", "IP_address", "File_path"],
+    "Digital & Technical Entities": ["URL", "Website", "Hashtag", "Email", "IP_address", "File_path"],
 }
 
 # --- Model Loading ---