Update src/streamlit_app.py

src/streamlit_app.py

@@ -82,12 +82,13 @@ 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
+	src="https://aiecosystem-hr-ai.hf.space"
+	frameborder="0"
+	width="850"
+	height="450"
     ></iframe>
+
     '''
     st.code(code, language="html")
     st.text("")
@@ -129,7 +130,7 @@ category_mapping = {
 def load_ner_model():
     """Loads the GLiNER model and caches it."""
     try:
-        return GLiNER.from_pretrained("gliner-community/gliner_xxl-v2.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
+        return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.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()
@@ -183,7 +184,7 @@ if st.button("Results"):
                         else:
                             st.info(f"No entities found for the '{category_name}' category.")
 
-                st.divider()
+                
 
                 with st.expander("See Glossary of tags"):
                     st.write('''