Update app.py
Browse files
app.py
CHANGED
|
@@ -15,7 +15,8 @@ def fill_mask_for_languages(sentences):
|
|
| 15 |
results = {}
|
| 16 |
for language, sentence in sentences.items():
|
| 17 |
masked_sentence = sentence.replace('____', unmasker.tokenizer.mask_token)
|
| 18 |
-
|
|
|
|
| 19 |
return results
|
| 20 |
|
| 21 |
def replace_mask(sentence, predicted_word):
|
|
|
|
| 15 |
results = {}
|
| 16 |
for language, sentence in sentences.items():
|
| 17 |
masked_sentence = sentence.replace('____', unmasker.tokenizer.mask_token)
|
| 18 |
+
unmasked = unmasker(masked_sentence)
|
| 19 |
+
results[language] = unmasked
|
| 20 |
return results
|
| 21 |
|
| 22 |
def replace_mask(sentence, predicted_word):
|