Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +1 -1
src/streamlit_app.py
CHANGED
|
@@ -176,7 +176,7 @@ category_mapping = {
|
|
| 176 |
def load_ner_model():
|
| 177 |
"""Loads the GLiNER model and caches it."""
|
| 178 |
try:
|
| 179 |
-
return GLiNER.from_pretrained("Ihor/gliner-biomed-large-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
|
| 180 |
except Exception as e:
|
| 181 |
st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
|
| 182 |
st.stop()
|
|
|
|
| 176 |
def load_ner_model():
|
| 177 |
"""Loads the GLiNER model and caches it."""
|
| 178 |
try:
|
| 179 |
+
return GLiNER.from_pretrained("Ihor/gliner-biomed-bi-large-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
|
| 180 |
except Exception as e:
|
| 181 |
st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
|
| 182 |
st.stop()
|