AIEcosystem commited on
Commit
5ef7d0f
·
verified ·
1 Parent(s): 0062d79

Update src/streamlit_app.py

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