AIEcosystem commited on
Commit
f1477a3
·
verified ·
1 Parent(s): ab9a378

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -207,7 +207,7 @@ if 'df_ner' in st.session_state and not st.session_state.df_ner.empty:
207
  def load_gliner_model():
208
  """Initializes and caches the GLiNER model for QA."""
209
  try:
210
- return GLiNER.from_pretrained("knowledgator/gliner-multitask-v2.0", device="cpu")
211
  except Exception as e:
212
  st.error(f"Error loading the GLiNER model: {e}")
213
  st.stop()
 
207
  def load_gliner_model():
208
  """Initializes and caches the GLiNER model for QA."""
209
  try:
210
+ return GLiNER.from_pretrained("knowledgator/gliner-multitask-v1.0", device="cpu")
211
  except Exception as e:
212
  st.error(f"Error loading the GLiNER model: {e}")
213
  st.stop()