Spaces:
Running
Running
Update app.py
Browse files
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[
|
| 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]
|