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,
|