Update app.py

app.py

@@ -24,7 +24,7 @@ with st.form(key='my_form'):
 
     if submit_button:
       token_ids = tokenizer.encode(prompt, return_tensors='pt')
-      token_ids_tk = tokenizer.tokenize(sentence, return_tensors='pt')
+      token_ids_tk = tokenizer.tokenize(prompt, return_tensors='pt')
       masked_position = (token_ids.squeeze() == tokenizer.mask_token_id).nonzero()
       masked_pos = [mask.item() for mask in masked_position ]
       with torch.no_grad():