pr-ent commited on
Commit
e88235f
1 Parent(s): f35867d

Fix mask bug

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -63,6 +63,7 @@ text = choose_text_menu('')
63
  for prompt in prompt_mix_list:
64
  model_load_state = st.text('Tagging Running')
65
  prompt = prompt.replace('[MASK]', '{}')
 
66
  output_tokens = query_model_prompting(model_prompting, text, prompt, top_k, targets=None)
67
  list_results = []
68
  for each in output_tokens:
63
  for prompt in prompt_mix_list:
64
  model_load_state = st.text('Tagging Running')
65
  prompt = prompt.replace('[MASK]', '{}')
66
+ prompt = prompt.format(model_prompting.tokenizer.mask_token)
67
  output_tokens = query_model_prompting(model_prompting, text, prompt, top_k, targets=None)
68
  list_results = []
69
  for each in output_tokens: