AIEcosystem commited on
Commit
435229b
·
verified ·
1 Parent(s): 6e2cafb

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -243,7 +243,7 @@ category_mapping = {
243
  def load_ner_model():
244
  """Loads the GLiNER model and caches it."""
245
  try:
246
- return GLiNER.from_pretrained("gretelai/gretel-gliner-bi-large-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
247
  except Exception as e:
248
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
249
  st.stop()
 
243
  def load_ner_model():
244
  """Loads the GLiNER model and caches it."""
245
  try:
246
+ return GLiNER.from_pretrained("urchade/gliner_multi_pii-v1", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
247
  except Exception as e:
248
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
249
  st.stop()