Update src/streamlit_app.py

src/streamlit_app.py

@@ -82,7 +82,12 @@ For any errors or inquiries, please contact us at [email protected]""")
 with st.sidebar:
     st.write("Use the following code to embed the HR.ai web app on your website. Feel free to adjust the width and height values to fit your page.")
     code = '''
-    <iframe	src="https://aiecosystem-hr-ai.hf.space"	frameborder="0"	width="850"	height="450"></iframe>
+    <iframe	
+    src="https://aiecosystem-hr-ai.hf.space"	
+    frameborder="0"	
+    width="850"	
+    height="450"
+    ></iframe>
     '''
     st.code(code, language="html")
     st.text("")
@@ -124,7 +129,7 @@ category_mapping = {
 def load_ner_model():
     """Loads the GLiNER model and caches it."""
     try:
-        return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
+        return GLiNER.from_pretrained("gliner-community/gliner_xxl-v2.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
     except Exception as e:
         st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
         st.stop()
@@ -141,7 +146,7 @@ def clear_text():
     st.session_state['my_text_area'] = ""
 
 st.button("Clear text", on_click=clear_text)
-st.divider()
+
 
 # --- Results Section ---
 if st.button("Results"):