Spaces:

pr-ent
/

pr-ent-dashboard

Runtime error

pr-ent commited on Jun 7, 2022

Commit

e88235f

1 Parent(s): f35867d

Fix mask bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -63,6 +63,7 @@ text = choose_text_menu('')
 for prompt in prompt_mix_list:
     model_load_state = st.text('Tagging Running')
     prompt = prompt.replace('[MASK]', '{}')
     output_tokens = query_model_prompting(model_prompting, text, prompt, top_k, targets=None)
     list_results = []
     for each in output_tokens:

 for prompt in prompt_mix_list:
     model_load_state = st.text('Tagging Running')
     prompt = prompt.replace('[MASK]', '{}')
+    prompt = prompt.format(model_prompting.tokenizer.mask_token)
     output_tokens = query_model_prompting(model_prompting, text, prompt, top_k, targets=None)
     list_results = []
     for each in output_tokens: