Update app.py

app.py

@@ -59,6 +59,7 @@ def setModel(model_checkpoint, aggregation):
     tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
     return pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy=aggregation)
 
+@st.cache(allow_output_mutation=True)
 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", " ")