Update src/streamlit_app.py

src/streamlit_app.py

@@ -105,7 +105,7 @@ def load_gliner_model(model_name):
     """Initializes and caches the GLiNER model."""
     try:
         if model_name == "HR_AI":
-            return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
+            return GLiNER.from_pretrained("gretelai/gretel-gliner-bi-large-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
         elif model_name == "InfoFinder":
             return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
     except Exception as e:
@@ -113,13 +113,13 @@ def load_gliner_model(model_name):
         st.stop()
 
 # --- HR_AI Model Labels and Mappings ---
-labels = ["Email", "Phone_number", "Street_address", "City", "Country", "Date_of_birth", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Terminated", "Retired", "Job_title", "Date", "Organization", "Role", "Performance_score", "Leave_of_absence", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance", "Pay_rate", "Annual_salary", "Tax", "Deductions", "Interview_type", "Applicant", "Referral", "Job_board", "Recruiter", "Offer_letter", "Agreement", "Certification", "Skill"]
+labels = ["Job_with_Date", "Email", "Phone_number", "Street_address", "City", "Country", "Date_of_birth", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Terminated", "Retired", "Job_title", "Date", "Organization", "Role", "Performance_score", "Leave_of_absence", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance", "Pay_rate", "Annual_salary", "Tax", "Deductions", "Interview_type", "Applicant", "Referral", "Job_board", "Recruiter", "Offer_letter", "Agreement", "Certification", "Skill"]
 
 category_mapping = {
     "Contact Information": ["Email", "Phone_number", "Street_address", "City", "Country"],
     "Personal Details": ["Date_of_birth", "Marital_status", "Person"],
     "Employment Status": ["Full_time", "Part_time", "Contract", "Terminated", "Retired"],
-    "Employment Information": ["Job_title", "Date", "Organization", "Role"],
+    "Employment Information": ["Job_with_Date", "Job_title", "Date", "Organization", "Role"],
     "Performance": ["Performance_score"],
     "Attendance": ["Leave_of_absence"],
     "Benefits": ["Retirement_plan", "Bonus", "Stock_options", "Health_insurance"],