Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +3 -2
src/streamlit_app.py
CHANGED
|
@@ -90,11 +90,12 @@ with st.sidebar:
|
|
| 90 |
st.write("Use the following code to embed the ChainSense web app on your website. Feel free to adjust the width and height values to fit your page.")
|
| 91 |
code = '''
|
| 92 |
<iframe
|
| 93 |
-
src="https://aiecosystem-
|
| 94 |
frameborder="0"
|
| 95 |
width="850"
|
| 96 |
height="450"
|
| 97 |
></iframe>
|
|
|
|
| 98 |
|
| 99 |
'''
|
| 100 |
st.code(code, language="html")
|
|
@@ -144,7 +145,7 @@ category_mapping = {
|
|
| 144 |
def load_ner_model():
|
| 145 |
"""Loads the GLiNER model and caches it."""
|
| 146 |
try:
|
| 147 |
-
return GLiNER.from_pretrained("gliner-
|
| 148 |
except Exception as e:
|
| 149 |
st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
|
| 150 |
st.stop()
|
|
|
|
| 90 |
st.write("Use the following code to embed the ChainSense web app on your website. Feel free to adjust the width and height values to fit your page.")
|
| 91 |
code = '''
|
| 92 |
<iframe
|
| 93 |
+
src="https://aiecosystem-chainsense.hf.space"
|
| 94 |
frameborder="0"
|
| 95 |
width="850"
|
| 96 |
height="450"
|
| 97 |
></iframe>
|
| 98 |
+
|
| 99 |
|
| 100 |
'''
|
| 101 |
st.code(code, language="html")
|
|
|
|
| 145 |
def load_ner_model():
|
| 146 |
"""Loads the GLiNER model and caches it."""
|
| 147 |
try:
|
| 148 |
+
return GLiNER.from_pretrained("knowledgator/gliner-multitask-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
|
| 149 |
except Exception as e:
|
| 150 |
st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
|
| 151 |
st.stop()
|