Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -43,9 +43,10 @@ languages = {
|
|
| 43 |
'Welsh': 'cym_Latn'
|
| 44 |
}
|
| 45 |
|
|
|
|
|
|
|
|
|
|
| 46 |
def translate_article(article, language):
|
| 47 |
-
tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
|
| 48 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
|
| 49 |
inputs = tokenizer(article, return_tensors="pt")
|
| 50 |
lang_code = languages[language]
|
| 51 |
translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code])
|
|
|
|
| 43 |
'Welsh': 'cym_Latn'
|
| 44 |
}
|
| 45 |
|
| 46 |
+
tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
|
| 47 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
|
| 48 |
+
|
| 49 |
def translate_article(article, language):
|
|
|
|
|
|
|
| 50 |
inputs = tokenizer(article, return_tensors="pt")
|
| 51 |
lang_code = languages[language]
|
| 52 |
translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id[lang_code])
|