Update pages/fill_mask.py

pages/fill_mask.py

@@ -13,7 +13,7 @@ def app():
 
     @st.cache(allow_output_mutation=True, suppress_st_warning =True, show_spinner=False)
     def get_model():
-        return pipeline('fill-mask', model='bert-base-uncased',skip_special_tokens=True)
+        return pipeline('fill-mask', model='bert-base-uncased')
 
     def create_graph(answer):
         x_bar = [i['token_str'] for i in answer]
@@ -32,7 +32,7 @@ def app():
         st.altair_chart(chart, use_container_width=True)
 
         
-    col1, col2 = st.beta_columns([2,1])
+    col1, col2 = st.columns([2,1])
 
 
     with col1: