Update src/streamlit_app.py

src/streamlit_app.py

@@ -74,8 +74,8 @@ st.link_button("by nlpblogs", "https://nlpblogs.com", type="tertiary")
 
 expander = st.expander("**Important notes on the Human Resources**")
 expander.write("""
-   **Named Entities:** This HR.ai predicts twenty-four (24) labels:
-"Email_address", "Phone_number", "Street_address", "City", "State", "Zip_code",
+   **Named Entities:** This HR.ai predicts sixty (60) labels:
+"Email_address", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country",
 "Date_of_birth", "Gender", "Marital_status", "Full_name",
 "Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired",
 "Job_title", "Employment_type", "Start_date", "End_date", "Company", "Organization", "Role", "Position",
@@ -137,7 +137,7 @@ if not comet_initialized:
 
 # --- Label Definitions ---
 labels = [
-"Email_Address", "Phone_Number", "Street_Address", "City", "State", "Zip_code",
+"Email_Address", "Phone_Number", "Street_Address", "City", "State", "Zip_code", "Country",
 "Date_of_Birth", "Gender", "Marital_Status", "Full_Name",
 "Full_Time", "Part_Time", "Contract", "Temporary", "Terminated", "Active", "Retired",
 "Job_Title", "Employment_Type", "Start_Date", "End_Date", "Company_Name", "Organization_Name", "Role", "Position",
@@ -157,7 +157,7 @@ labels = [
 category_mapping = {
 
     
-    "Contact Information": ["Email_Address", "Phone_Number", "Street_Address", "City", "State", "Zip_code"],
+    "Contact Information": ["Email_Address", "Phone_Number", "Street_Address", "City", "State", "Zip_code", "Country"],
     "Personal Details": ["Date of birth", "Gender", "Marital_Status", "Full_Name"],
     "Employment Status": ["Full_Time", "Part_Time", "Contract", "Temporary", "Terminated", "Active", "Retired"],
     "Employment Information" : ["Job_Title", "Employment_Type", "Start_Date", "End_Date", "Company_Name", "Organization_Name", "Role", "Position"],