Spaces:

nassga
/

BiomedicalNER

Sleeping

nassga commited on Jun 20, 2023

Commit

6323cf9

1 Parent(s): 0856724

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ ENTITY_TO_COLOR = {
     'MISC': '#fea',
 }
-@st.cache(allow_output_mutation=True, show_spinner=False)
 def get_pipe():
     model_name = "dslim/bert-base-NER"
     model = transformers.AutoModelForTokenClassification.from_pretrained(model_name)

     'MISC': '#fea',
 }
+@cache_data(allow_output_mutation=True, show_spinner=False)
 def get_pipe():
     model_name = "dslim/bert-base-NER"
     model = transformers.AutoModelForTokenClassification.from_pretrained(model_name)