Update app.py

app.py

@@ -18,8 +18,13 @@ st.set_page_config(
 
 @st.cache(show_spinner=False, allow_output_mutation=True, suppress_st_warning=True)
 def load_models():
-    spanish_model = spacy.load("./spacy-models/toy_output_es_blank/model-best/")
-    english_model = spacy.load("./spacy-models/toy_output_en_blank/model-best/")
+    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")            
+        
+    if selected_for == "Efficiency":
+        spanish_model = spacy.load("./spacy-models/toy_output_es_blank/model-best")
+        english_model = spacy.load("./spacy-models/toy_output_en_blank/model-best/")     
     models = {"English": english_model, "Spanish": spanish_model}
     return models
 
@@ -52,57 +57,72 @@ def process_text(doc, selected_multi_ner):
 # Side bar
 selected_language = st.sidebar.selectbox("Select a language", options=["English", "Spanish"])
 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'])
 
 models = load_models()
 selected_model = models[selected_language]
 
 import base64
 
-LOGO_IMAGE = "aida_logo.png"
+LOGO_IMAGE = "LeetSpeak-NER-cropped.png"
 
 st.markdown(
     """
     <style>
-    .container {
-        display: flex;
-
-    }
     .logo-img {
-        float:right;
-        margin-top: 2.2em;
-        margin-left: -10em;
+
+
+        margin-top: auto;
+        margin-left: 30%;
+        width: 30%;
+        
     }
+    .logo-img-2 {
+    margin-top: 10%;
+      margin-left: 20%;
+      width: 35%;        
+        
+    }   
     </style>
     """,
     unsafe_allow_html=True
 )
 
 
-col1, col2 = st.columns([4, 1])
-with col1: 
-    st.markdown("""
-    <style>
-    .big-font {
-        font-size:3em;
-        font-weight: bold;
-    }
-    </style>
-    """, unsafe_allow_html=True)
-
-    st.markdown('<p class="big-font">Welcome to <font color="#4B8BBE">Leet</font><font color="#FFE873">Speak</font><font color="#ff73a2">-NER</font></p>', unsafe_allow_html=True)
-with col2:
+col1, col2= st.columns([2, 2])
+with col1:
     # st.image('./aida_logo.png')
      st.markdown(
         f"""
-        <div class="container">
             <img class="logo-img" src="data:image/png;base64,{base64.b64encode(open(LOGO_IMAGE, "rb").read()).decode()}">
-        </div>
         """,
         unsafe_allow_html=True
     )  
+    
+with col2:
+    # st.image('./aida_logo.png')
+     st.markdown(
+        f"""
+            <img class="logo-img-2" src="data:image/png;base64,{base64.b64encode(open("aida_logo.png", "rb").read()).decode()}">
+        """,
+        unsafe_allow_html=True
+    )  
+
+# st.image([LOGO_IMAGE,"aida_logo.png"], width=100)
 
 
+st.markdown("""
+    <style>
+    .big-font {
+        font-size:3em;
+        font-weight: bold;
+    }
+    </style>
+    """, unsafe_allow_html=True)
 
+st.markdown('<p class="big-font">Welcome to <font color="#4B8BBE">Leet</font><font color=" #FFD43B">Speak</font><font color="#ff73a2">-NER</font></p>', unsafe_allow_html=True)
+        
+        
 with st.expander("Project Description", expanded=False):
      st.write("""
          Developed in Applied Intelligence and Data Analysis ([AI+DA](http://aida.etsisi.upm.es/)) group at Polytech University of Madrid (UPM).
@@ -110,9 +130,29 @@ with st.expander("Project Description", expanded=False):
          
          Currently, two languages are supported: English and Spanish. Additionally, you can select whether the detected entities are broken down into the three types of camouflaged words: Canonical Leetspeak, Punctuation Camouflaged, Inversion Camouflaged.
      """)  
+        
+with st.expander("Try some of these examples", expanded=False):
+     st.write("""
+    ENGLISH:
+    - Desperately dominated by fam1ly sitüatløns, he leaves her.
+    - You might as well come out to investigate a strang3 n'o?i+se or something.
+    - But one other thing that we have to re;think is the way that we dy£ our #c!l.o|th?£+s.
+    - And he wanted Baltimore to get that same kind of att£ntløn from the outside, but )i)n)t)r)o)s)p)e)c)t)i)o)n from the inside about what was going on with us.
+
+     
+    SPANISH
+    - _d+i%o"s mío!
+    - se asocian con el m13;d0 y el d'o'lor. g£rønlmo solía decir
+    - Con las nuevas tecnologías digitales, los agrlcultør£s pueden manejar mejor el uso de sus tierras, su energía y su agua, y prepararse para el mal clima.
+    - En el tiempo transcurrido entre mi período de escuela %s%3%c%_%n%d%a%r%1%a y el mo'm3n'to de empezar a enseñar vimos surgir el fenómeno de in't£r'net
+    - Las pre0c_pac1on3s van desde inquietudes por las ramificaciones desestabilizadoras de una estrategia de salida de la FC, hasta aprehensión por pérdidas de capital en la rápidamente creciente cartera de valores de la Fed (actualmente de $3 billones y en camino a los $4 billones para finales de este a��o).
+     """)  
     
+    #     - Why do all these _r_e_p_o_r_t_e_r_s, who get praise and money for doing what Assange has done, maintain a cow;ardly silence (at best) while a fellow publisher faces threats of extradition, banning, and espionage charges (which can incur the death penalty), not to mention calls for his as'sa'ss1nat'i'on?
     
-    
+    # - Cada uno de estos es un crimen de guerra, un crimen contra la humanidad y, en el caso de los asesinatos masivos de la campaña de Anfal, y tal vez también en el caso de los árabes de los pantanos, el crimen más serio de todos, ge'no'ci'dio.    
+  
+    # - No quiere decir que debamos iniciar una campaña por los derechos de los lns£ctøs
     
 
 st.subheader("Input Text")   
@@ -135,8 +175,9 @@ with st.form("my_form"):
 
     
 st.subheader("Output")
-doc = selected_model(text_input)
-tokens = process_text(doc, selected_multi_ner)
+with st.spinner('Wait for it...'):
+    doc = selected_model(text_input)
+    tokens = process_text(doc, selected_multi_ner)
 
-annotated_text(*tokens)
+    annotated_text(*tokens)