Spaces:
Runtime error
Runtime error
Fix mask bug
Browse files
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:
|