Spaces:
Runtime error
Runtime error
Commit
·
4472bb8
1
Parent(s):
04d97c8
Debug indicberta
Browse files- apps/inference.py +1 -0
apps/inference.py
CHANGED
|
@@ -18,6 +18,7 @@ def load_model(masked_text, model_name):
|
|
| 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
|
|
|
|
| 18 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 19 |
MASK_TOKEN = tokenizer.mask_token
|
| 20 |
masked_text = masked_text.replace("<mask>", MASK_TOKEN)
|
| 21 |
+
print(model_name, MASK_TOKEN, masked_text)
|
| 22 |
nlp = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 23 |
result_sentence = nlp(masked_text)
|
| 24 |
return result_sentence
|