Update app.py

app.py

@@ -20,7 +20,7 @@ st.set_page_config(
 def load_models():
     if selected_for == "Accuracy":
         spanish_model = spacy.load("./spacy-models/output_full_ES_roberta-base-bne/model-best")
-        english_model = spacy.load("./spacy-models/output_full_EN_roberta_base/model-best")            
+        english_model = spacy.load("./spacy-models/output_full_EN_roberta_base/model-best") 
         
     if selected_for == "Efficiency":
         spanish_model = spacy.load("./spacy-models/toy_output_es_blank/model-best")
@@ -28,6 +28,10 @@ def load_models():
     models = {"English": english_model, "Spanish": spanish_model}
     return models
 
+@st.cache(show_spinner=True, allow_output_mutation=True, suppress_st_warning=True)
+def load_xx_model():
+    return  spacy.load("xx_LeetSpeakNER_mstsb_mpnet")
+    
 
 # 'INV_CAMO', 'LEETSPEAK', 'MIX', 'PUNCT_CAMO'
 def process_text(doc, selected_multi_ner):
@@ -55,12 +59,15 @@ def process_text(doc, selected_multi_ner):
 
 
 # Side bar
-selected_language = st.sidebar.selectbox("Select a language", options=["English", "Spanish"])
+# selected_language = st.sidebar.selectbox("Select a language", options=["English", "Spanish"])
+selected_language = st.sidebar.selectbox("Select a language", options=["Multilingual"])
 selected_multi_ner = st.sidebar.radio('Do you want to break down the Entities detected by type of leetspeak?', ['Yes', 'No'])
-selected_for = st.sidebar.radio('Select for:', ['Efficiency', 'Accuracy'])
+# selected_for = st.sidebar.radio('Select for:', ['Efficiency', 'Accuracy'])
+# models = load_models()
+# selected_model = models[selected_language]
+
 
-models = load_models()
-selected_model = models[selected_language]
+selected_model = load_xx_model()
 
 import base64