Update app.py

app.py

@@ -24,8 +24,9 @@ def decode(tokenizer, pred_idx, top_clean):
   tokens = []
   for w in pred_idx:
     token = ''.join(tokenizer.decode(w).split())
-    if token not in ignore_tokens:
-      tokens.append(token.replace('##', ''))
+    #if token not in ignore_tokens:
+    #  tokens.append(token.replace('##', ''))
+    tokens.append(token)
   return '\n'.join(tokens[:top_clean])
 
 def encode(tokenizer, text_sentence, add_special_tokens=True):
@@ -74,7 +75,7 @@ try:
             
       model_name = st.sidebar.selectbox(label='Select Model to Apply',  options=['ajitrajasekharan/biomedical', 'bert-base-cased','bert-large-cased','microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract-fulltext','allenai/scibert_scivocab_cased'], index=0,  key = "model_name")
       bert_tokenizer, bert_model  = load_bert_model(model_name)
-      default_text = "Imatinib is used to treat"
+      default_text = "Imatinib is used to [MASK] nsclc"
       input_text = st.text_area(
                   label="Original text",
                   value=default_text,