Update src/streamlit_app.py

src/streamlit_app.py

@@ -75,7 +75,7 @@ st.link_button("by nlpblogs", "https://nlpblogs.com", type="tertiary")
 expander = st.expander("**Important notes**")
 expander.write("""
    **Named Entities:** This HR.ai predicts sixty (60) labels:
-"Email", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country", "Date_of_birth", "Gender", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired", "Job_title", "Employment_type", "Start_date", "End_date", "Company", "Organization", "Role", "Position",
+"Email", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country", "Date_of_birth", "Gender", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired", "Job_title", "Employment_type", "Year", "Date", "Company", "Organization", "Role", "Position",
 "Performance_review", "Performance_rating", "Performance_score", "Sick_days", "Vacation_days", "Leave_of_absence", "Holidays", "Pension", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance","Retire_date",
 "Pay_rate", "Hourly_wage", "Annual_salary", "Overtime_pay", "Tax", "Social_security", "Deductions", "Job_posting", "Job_description", "Interview_type", "Applicant", "Candidate", "Referral", "Job_board", "Recruiter",
 "Contract", "Offer_letter", "Agreement", "Training_course", "Certification", "Skill"
@@ -132,7 +132,7 @@ labels = [
 "Email", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country",
 "Date_of_birth", "Gender", "Marital_status", "Person",
 "Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired",
-"Job_title", "Employment_type", "Start_date", "End_date", "Company", "Organization", "Role", "Position",
+"Job_title", "Employment_type", "Year", "Date", "Company", "Organization", "Role", "Position",
 "Performance_review", "Performance_rating", "Performance_score",
 "Sick_days", "Vacation_days", "Leave_of_absence", "Holidays",
 "Pension", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance","Retire_date",
@@ -152,7 +152,7 @@ category_mapping = {
     "Contact Information": ["Email", "Phone_number", "Street_address", "City", "State", "Zip_code", "Country"],
     "Personal Details": ["Date_of_birth", "Gender", "Marital_status", "Person"],
     "Employment Status": ["Full_time", "Part_time", "Contract", "Temporary", "Terminated", "Active", "Retired"],
-    "Employment Information" : ["Job_title", "Employment_type", "Start_date", "End_date", "Company", "Organization", "Role", "Position"],
+    "Employment Information" : ["Job_title", "Employment_type", "Year", "Date", "Company", "Organization", "Role", "Position"],
     
     "Performance": ["Performance_review", "Performance_rating", "Performance_score"],
     "Attendance": ["Sick_days", "Vacation_days", "Leave_of_absence", "Holidays"],