Update app.py

app.py

@@ -33,6 +33,9 @@ def decode(tokenizer, pred_idx, top_clean):
   return '\n'.join(tokens[:top_clean])
 
 def encode(tokenizer, text_sentence, add_special_tokens=True):
+  bert_tokenizer = st.session_state['bert_tokenizer']
+  bert_model = st.session_state['bert_model']
+  
   text_sentence = text_sentence.replace('<mask>', tokenizer.mask_token)
     # if <mask> is the last token, append a "." so that models dont predict punctuation.
   #if tokenizer.mask_token == text_sentence.split()[-1]:
@@ -47,6 +50,9 @@ def encode(tokenizer, text_sentence, add_special_tokens=True):
   return input_ids, mask_idx,tokenized_text
 
 def get_all_predictions(text_sentence, model_name,top_clean=5):
+  bert_tokenizer = st.session_state['bert_tokenizer']
+  bert_model = st.session_state['bert_model']
+  
     # ========================= BERT =================================
   input_ids, mask_idx,tokenized_text = encode(bert_tokenizer, text_sentence)
    
@@ -95,7 +101,7 @@ def on_text_change():
 def on_option_change():
 
   text = st.session_state.my_choice
-  st.info("Preselected text chosen")
+  st.info("Preselected text chosen:" + text)
   run_test(text,st.session_state['top_k'],st.session_state['model_name'])
   
 def on_results_count_change():