seanp03 commited on
Commit
8183d04
·
verified ·
1 Parent(s): dd77d0b

fix pt and nl mbart calls

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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,