Update app.py
Browse files
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(
|
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]]
|