rebuild

app.py

@@ -118,6 +118,8 @@ col1.image(f"{statics}/logo_cnrs.png", width=80)
 # ===== SOME FUNCTIONS =====
 
 
+# Cached to prevent computation on every rerun
+@st.cache_resource
 def download_sentencizer():
     """
     Downloads the fr_arches_sentencizer model.
@@ -133,6 +135,8 @@ def download_sentencizer():
         subprocess.run(["pip", "install", "https://huggingface.co/a-menu/fr_arches_sentencizer/resolve/main/fr_arches_sentencizer-any-py3-none-any.whl"])
 
 
+# Cached to prevent computation on every rerun
+@st.cache_resource
 def download_ner_trf():
     """
     Downloads the fr_arches_ner_trf TRF NER model.
@@ -148,6 +152,8 @@ def download_ner_trf():
         subprocess.run(["pip", "install", "https://huggingface.co/a-menu/fr_arches_ner_trf/resolve/main/fr_arches_ner_trf-any-py3-none-any.whl"])
 
 
+# Cached to prevent computation on every rerun
+@st.cache_resource
 def download_ner():
     """
     Downloads the fr_arches_ner NER model.
@@ -163,6 +169,8 @@ def download_ner():
         subprocess.run(["pip", "install", "https://huggingface.co/a-menu/fr_arches_ner/resolve/main/fr_arches_ner-any-py3-none-any.whl"])
 
 
+# Cached to prevent computation on every rerun
+@st.cache_resource
 def load_sentencizer():
     """
     Loads our custom sentence segmentation model.
@@ -175,6 +183,8 @@ def load_sentencizer():
     return senter
 
 
+# Cached to prevent computation on every rerun
+@st.cache_resource
 def load_ner_trf():
     """
     Loads our custom fr_arches_ner_trf trf ner model.
@@ -190,6 +200,8 @@ def load_ner_trf():
     return ner
 
 
+# Cached to prevent computation on every rerun
+@st.cache_resource
 def load_ner():
     """
     Loads our custom fr_arches_ner ner model.