TiberiuCristianLeon commited on
Commit
44aa6cb
·
verified ·
1 Parent(s): 4746bc2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def translate_text(input_text, sselected_language, tselected_language, model_nam
38
  tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
39
  # translate source to target
40
  tokenizer.src_lang = mbart_large_languages[sselected_language]
41
- encoded = tokenizer(article_hi, return_tensors="pt")
42
  generated_tokens = model.generate(
43
  **encoded,
44
  forced_bos_token_id=tokenizer.lang_code_to_id[mbart_large_languages[tselected_language]]
 
38
  tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
39
  # translate source to target
40
  tokenizer.src_lang = mbart_large_languages[sselected_language]
41
+ encoded = tokenizer(input_text, return_tensors="pt")
42
  generated_tokens = model.generate(
43
  **encoded,
44
  forced_bos_token_id=tokenizer.lang_code_to_id[mbart_large_languages[tselected_language]]