Update src/streamlit_app.py

src/streamlit_app.py

@@ -75,7 +75,7 @@ def build_albertina_page():
 
     top_k = st.sidebar.number_input('Number of predictions to return', min_value=1, max_value=5, value=1, step=1)
 
-    st.write("Enter a sentence with a **<mask>** token, and the model will predict the missing word.")
+    st.write("Enter a sentence with a **[MASK]** token, and the model will predict the missing word.")
 
     results = None
 
@@ -90,7 +90,7 @@ def build_albertina_page():
             if submit and input_text:
                 results = instantiate_albertina(top_k, input_text)
         except Exception as e:
-            st.warning('There must be a special token "<mask>" in sentence!', icon="⚠️")
+            st.warning('There must be a special token "[MASK]" in sentence!', icon="⚠️")
             st.warning(e)
 
     with col2: