fix pt and nl mbart calls
Browse files
app.py
CHANGED
|
@@ -112,6 +112,7 @@ async def translate(request: Request):
|
|
| 112 |
if is_mbart:
|
| 113 |
# build e.g. "de_DE", "es_XX", etc.
|
| 114 |
lang_code = f"{target_lang}_{target_lang.upper()}"
|
|
|
|
| 115 |
bos_id = tokenizer.lang_code_to_id[lang_code]
|
| 116 |
outputs = model.generate(
|
| 117 |
**inputs,
|
|
|
|
| 112 |
if is_mbart:
|
| 113 |
# build e.g. "de_DE", "es_XX", etc.
|
| 114 |
lang_code = f"{target_lang}_{target_lang.upper()}"
|
| 115 |
+
if target_lang == "nl" or target_lang == "pt": lang_code = f"{target_lang}_XX"
|
| 116 |
bos_id = tokenizer.lang_code_to_id[lang_code]
|
| 117 |
outputs = model.generate(
|
| 118 |
**inputs,
|