AIEcosystem commited on
Commit
bb99ec1
·
verified ·
1 Parent(s): 97d4895

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -107,7 +107,7 @@ def load_gliner_model(model_name):
107
  if model_name == "HR_AI":
108
  return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
109
  elif model_name == "InfoFinder":
110
- return GLiNER.from_pretrained("knowledgator/gliner-multitask-v1.0", device="cpu")
111
  except Exception as e:
112
  st.error(f"Error loading the GLiNER model: {e}")
113
  st.stop()
 
107
  if model_name == "HR_AI":
108
  return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", nested_ner=True, num_gen_sequences=2, gen_constraints=labels)
109
  elif model_name == "InfoFinder":
110
+ return GLiNER.from_pretrained("knowledgator/gliner-multitask-large-v0.5", device="cpu")
111
  except Exception as e:
112
  st.error(f"Error loading the GLiNER model: {e}")
113
  st.stop()