Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +8 -7
src/streamlit_app.py
CHANGED
|
@@ -82,12 +82,13 @@ For any errors or inquiries, please contact us at info@nlpblogs.com""")
|
|
| 82 |
with st.sidebar:
|
| 83 |
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.")
|
| 84 |
code = '''
|
| 85 |
-
<iframe
|
| 86 |
-
|
| 87 |
-
|
| 88 |
-
|
| 89 |
-
|
| 90 |
></iframe>
|
|
|
|
| 91 |
'''
|
| 92 |
st.code(code, language="html")
|
| 93 |
st.text("")
|
|
@@ -129,7 +130,7 @@ category_mapping = {
|
|
| 129 |
def load_ner_model():
|
| 130 |
"""Loads the GLiNER model and caches it."""
|
| 131 |
try:
|
| 132 |
-
return GLiNER.from_pretrained("gliner-
|
| 133 |
except Exception as e:
|
| 134 |
st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
|
| 135 |
st.stop()
|
|
@@ -183,7 +184,7 @@ if st.button("Results"):
|
|
| 183 |
else:
|
| 184 |
st.info(f"No entities found for the '{category_name}' category.")
|
| 185 |
|
| 186 |
-
|
| 187 |
|
| 188 |
with st.expander("See Glossary of tags"):
|
| 189 |
st.write('''
|
|
|
|
| 82 |
with st.sidebar:
|
| 83 |
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.")
|
| 84 |
code = '''
|
| 85 |
+
<iframe
|
| 86 |
+
src="https://aiecosystem-hr-ai.hf.space"
|
| 87 |
+
frameborder="0"
|
| 88 |
+
width="850"
|
| 89 |
+
height="450"
|
| 90 |
></iframe>
|
| 91 |
+
|
| 92 |
'''
|
| 93 |
st.code(code, language="html")
|
| 94 |
st.text("")
|
|
|
|
| 130 |
def load_ner_model():
|
| 131 |
"""Loads the GLiNER model and caches it."""
|
| 132 |
try:
|
| 133 |
+
return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
|
| 134 |
except Exception as e:
|
| 135 |
st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
|
| 136 |
st.stop()
|
|
|
|
| 184 |
else:
|
| 185 |
st.info(f"No entities found for the '{category_name}' category.")
|
| 186 |
|
| 187 |
+
|
| 188 |
|
| 189 |
with st.expander("See Glossary of tags"):
|
| 190 |
st.write('''
|