Update app.py

app.py

@@ -34,8 +34,8 @@ def encode(tokenizer, text_sentence, add_special_tokens=True):
   if tokenizer.mask_token == text_sentence.split()[-1]:
     text_sentence += ' .'
 
-    input_ids = torch.tensor([tokenizer.encode(text_sentence, add_special_tokens=add_special_tokens)])
-    mask_idx = torch.where(input_ids == tokenizer.mask_token_id)[1].tolist()[0]
+  input_ids = torch.tensor([tokenizer.encode(text_sentence, add_special_tokens=add_special_tokens)])
+  mask_idx = torch.where(input_ids == tokenizer.mask_token_id)[1].tolist()[0]
   return input_ids, mask_idx
 
 def get_all_predictions(text_sentence, top_clean=5):
@@ -48,7 +48,7 @@ def get_all_predictions(text_sentence, top_clean=5):
 
 def get_bert_prediction(input_text,top_k):
   try:
-    input_text += ' <mask>'
+    #input_text += ' <mask>'
     res = get_all_predictions(input_text, top_clean=int(top_k))
     return res
   except Exception as error: