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]
|