Spaces:
Runtime error
Runtime error
Commit
·
04d97c8
1
Parent(s):
c1b0837
Fixed typo
Browse files- apps/inference.py +1 -1
apps/inference.py
CHANGED
|
@@ -17,7 +17,7 @@ def load_model(masked_text, model_name):
|
|
| 17 |
model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=from_flax)
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 19 |
MASK_TOKEN = tokenizer.mask_token
|
| 20 |
-
masked_text = masked_text.
|
| 21 |
nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 22 |
result_sentence = nlp(masked_text)
|
| 23 |
return result_sentence
|
|
|
|
| 17 |
model = AutoModelForMaskedLM.from_pretrained(model_name, from_flax=from_flax)
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 19 |
MASK_TOKEN = tokenizer.mask_token
|
| 20 |
+
masked_text = masked_text.replace("<mask>", MASK_TOKEN)
|
| 21 |
nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 22 |
result_sentence = nlp(masked_text)
|
| 23 |
return result_sentence
|