AIEcosystem commited on
Commit
e275008
·
verified ·
1 Parent(s): af7f246

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -156,7 +156,7 @@ category_mapping = {
156
  def load_ner_model():
157
  """Loads the GLiNER model and caches it."""
158
  try:
159
- return GLiNER.from_pretrained("EmergentMethods/gliner_medium_news-v2.1", nested_ner=True, num_gen_sequences=2, gen_constraints= labels, threshold = 0.70)
160
  except Exception as e:
161
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
162
  st.stop()
 
156
  def load_ner_model():
157
  """Loads the GLiNER model and caches it."""
158
  try:
159
+ return GLiNER.from_pretrained("EmergentMethods/gliner_large_news-v2.1", nested_ner=True, num_gen_sequences=2, gen_constraints= labels, threshold = 0.70)
160
  except Exception as e:
161
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
162
  st.stop()