TiberiuCristianLeon commited on
Commit
31da34a
·
verified ·
1 Parent(s): 3e539e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -207,7 +207,7 @@ def mbart_many_to_one(model_name, sl, tl, input_text):
207
  model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
208
  tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
209
  # translate to English
210
- tokenizer.src_lang = languagecodes.mbart_large_languages[tl]
211
  encoded = tokenizer(input_text, return_tensors="pt")
212
  generated_tokens = model.generate(**encoded)
213
  return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
 
207
  model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
208
  tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-one-mmt")
209
  # translate to English
210
+ tokenizer.src_lang = languagecodes.mbart_large_languages[sl]
211
  encoded = tokenizer(input_text, return_tensors="pt")
212
  generated_tokens = model.generate(**encoded)
213
  return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]