TheBeast commited on
Commit
285a100
1 Parent(s): f3c5980

Changing streamlit cache decorator

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from simpletransformers.ner import NERModel, NERArgs
16
  HTML_WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
17
 
18
 
19
- @st.cache_resource
20
  def load_models():
21
  """Load custom built NER4OPT model."""
22
  custom_labels = [
 
16
  HTML_WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
17
 
18
 
19
+ @st.cache(allow_output_mutation=true)
20
  def load_models():
21
  """Load custom built NER4OPT model."""
22
  custom_labels = [