Update app.py

app.py

@@ -53,10 +53,17 @@ elif input_method == "Write or Paste New Text":
     st.subheader("Text to Run")
     input_text = st.text_area('Write or Paste Text Below', value="", height=128, max_chars=None, key=2)
 
+@st.cache
 def setModel(model_checkpoint, aggregation):
     model = AutoModelForTokenClassification.from_pretrained(model_checkpoint)
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
     return pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy=aggregation)
+    
+@st.cache
+def get_html(html: str):
+    WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
+    html = html.replace("\n", " ")
+    return WRAPPER.format(html)
         
 Run_Button = st.button("Run", key=None)
 if Run_Button == True:
@@ -76,11 +83,6 @@ if Run_Button == True:
     st.subheader("Recognized Entities")
     st.dataframe(df_final)
 
-    def get_html(html: str):
-        WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
-        html = html.replace("\n", " ")
-        return WRAPPER.format(html)
-    
     st.subheader("Spacy Style Display")
 
     spacy_display = {}