AIEcosystem commited on
Commit
24e215e
·
verified ·
1 Parent(s): b22aae9

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -178,7 +178,7 @@ category_mapping = {
178
  def load_ner_model():
179
  """Loads the GLiNER model and caches it."""
180
  try:
181
- return GLiNER.from_pretrained("gliner-community/gliner_xxl-v2.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
182
 
183
  except Exception as e:
184
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
 
178
  def load_ner_model():
179
  """Loads the GLiNER model and caches it."""
180
  try:
181
+ return GLiNER.from_pretrained("gliner-community/gliner_large-v2.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
182
 
183
  except Exception as e:
184
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")