AIEcosystem commited on
Commit
adcc12e
·
verified ·
1 Parent(s): 39955bd

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -127,7 +127,7 @@ def load_gliner_model(model_name):
127
  """Initializes and caches the GLiNER model."""
128
  try:
129
  if model_name == "HR_AI":
130
- return GLiNER.from_pretrained("gretelai/gretel-gliner-bi-large-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
131
  elif model_name == "InfoFinder":
132
  return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
133
  except Exception as e:
 
127
  """Initializes and caches the GLiNER model."""
128
  try:
129
  if model_name == "HR_AI":
130
+ return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
131
  elif model_name == "InfoFinder":
132
  return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
133
  except Exception as e: