AIEcosystem commited on
Commit
35631ca
·
verified ·
1 Parent(s): b0e67fa

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -2
src/streamlit_app.py CHANGED
@@ -90,11 +90,12 @@ with st.sidebar:
90
  st.write("Use the following code to embed the ChainSense web app on your website. Feel free to adjust the width and height values to fit your page.")
91
  code = '''
92
  <iframe
93
- src="https://aiecosystem-entityfinance.hf.space"
94
  frameborder="0"
95
  width="850"
96
  height="450"
97
  ></iframe>
 
98
 
99
  '''
100
  st.code(code, language="html")
@@ -144,7 +145,7 @@ category_mapping = {
144
  def load_ner_model():
145
  """Loads the GLiNER model and caches it."""
146
  try:
147
- return GLiNER.from_pretrained("gliner-community/gliner_large-v2.5", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
148
  except Exception as e:
149
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
150
  st.stop()
 
90
  st.write("Use the following code to embed the ChainSense web app on your website. Feel free to adjust the width and height values to fit your page.")
91
  code = '''
92
  <iframe
93
+ src="https://aiecosystem-chainsense.hf.space"
94
  frameborder="0"
95
  width="850"
96
  height="450"
97
  ></iframe>
98
+
99
 
100
  '''
101
  st.code(code, language="html")
 
145
  def load_ner_model():
146
  """Loads the GLiNER model and caches it."""
147
  try:
148
+ return GLiNER.from_pretrained("knowledgator/gliner-multitask-v1.0", nested_ner=True, num_gen_sequences=2, gen_constraints= labels)
149
  except Exception as e:
150
  st.error(f"Failed to load NER model. Please check your internet connection or model availability: {e}")
151
  st.stop()