cache

demo.py

@@ -20,18 +20,26 @@ import streamlit as st
 
 # Load from huggingface
 # sm = spacy.load('en_core_web_sm', disable=['ner'])
+
+
+@st.cache(allow_output_mutation=True)
+def load_model(spacy_model):
+    nlp = spacy.load("en_engagement_RoBERTa_combined")
+    return (nlp)
+
+
 nlp = spacy.load("en_engagement_RoBERTa_combined")
 
 doc = nlp(
     'Welcome! Probably this is one of the few attempts to teach a machine how to read the discourse...! Although it is not perfect, you should be able to get a good place to start for your stance-taking analyses. The result will be presented here.'
 )
 
-TPL_ENT = """
-<mark class="entity" style="background: {bg}; padding: 0.45em 0.6em; margin: 0 0.25em; line-height: 1; border-radius: 0.35em;">
-    {text}
-    <span style="font-size: 0.8em; font-weight: bold; line-height: 1; border-radius: 0.35em; vertical-align: middle; margin-left: 0.5rem">{label}</span>
-</mark>
-"""
+# TPL_ENT = """
+# <mark class="entity" style="background: {bg}; padding: 0.45em 0.6em; margin: 0 0.25em; line-height: 1; border-radius: 0.35em;">
+#     {text}
+#     <span style="font-size: 0.8em; font-weight: bold; line-height: 1; border-radius: 0.35em; vertical-align: middle; margin-left: 0.5rem">{label}</span>
+# </mark>
+# """
 
 TPL_SPANS = """
 <div class="spans" style="line-height: 4.5;">
@@ -64,13 +72,13 @@ TPL_SPAN_START = """
 
 """
 
-TPL_SPAN_START_RTL = """
-<span style="background: {bg}; top: {top_offset}px; height: 4px; border-top-left-radius: 3px; border-bottom-left-radius: 3px; left: -1px; width: calc(100% + 2px); position: absolute;">
-    <span style="background: {bg}; z-index: 10; color: #000; top: -0.5em; padding: 2px 3px; position: absolute; font-size: 0.6em; font-weight: bold; line-height: 1; border-radius: 3px">
-        {label}{kb_link}
-    </span>
-</span>
-"""
+# TPL_SPAN_START_RTL = """
+# <span style="background: {bg}; top: {top_offset}px; height: 4px; border-top-left-radius: 3px; border-bottom-left-radius: 3px; left: -1px; width: calc(100% + 2px); position: absolute;">
+#     <span style="background: {bg}; z-index: 10; color: #000; top: -0.5em; padding: 2px 3px; position: absolute; font-size: 0.6em; font-weight: bold; line-height: 1; border-radius: 3px">
+#         {label}{kb_link}
+#     </span>
+# </span>
+# """
 
 DEFAULT_TEXT = """Tickner said regardless of the result, the royal commission was a waste of money and he would proceed with a separate inquiry into the issue headed by Justice Jane Matthews. His attack came as the Aboriginal women involved in the case demanded a female minister examine the religious beliefs they claim are inherent in their fight against a bridge to the island near Goolwa in South Australia."""