Update src/streamlit_app.py

src/streamlit_app.py

@@ -119,14 +119,14 @@ 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_address","IP_address","File_path"]
 
 # Corrected mapping dictionary
 
 # Create a mapping dictionary for labels to categories
 category_mapping = {
     "Core Foundational Entities": ["Person", "Organization", "Location", "Date", "Time", "Quantity"],
-    "Content Enrichment Entities": ["Product", "Event", "Title", "Job Title", "Artwork", "Media"],
+    "Content Enrichment Entities": ["Product", "Event", "Title", "Job_title", "Artwork", "Media"],
     "Digital & Technical Entities": ["URL", "Website", "Hashtag", "Email_address", "IP_address", "File_path"],
 }