Spaces:
Running
Running
Update src/Translate.py
Browse files- src/Translate.py +1 -1
src/Translate.py
CHANGED
|
@@ -39,7 +39,7 @@ class Translators:
|
|
| 39 |
def mbartlarge(self):
|
| 40 |
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
| 41 |
src_lang = f"{self.sl}_XX"
|
| 42 |
-
tgt_lang = f"{self.
|
| 43 |
# Load model and tokenizer
|
| 44 |
model = MBartForConditionalGeneration.from_pretrained(self.model_name)
|
| 45 |
tokenizer = MBart50TokenizerFast.from_pretrained(self.model_name, src_lang=src_lang)
|
|
|
|
| 39 |
def mbartlarge(self):
|
| 40 |
from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
| 41 |
src_lang = f"{self.sl}_XX"
|
| 42 |
+
tgt_lang = f"{self.tl}_{self.tl.upper()}"
|
| 43 |
# Load model and tokenizer
|
| 44 |
model = MBartForConditionalGeneration.from_pretrained(self.model_name)
|
| 45 |
tokenizer = MBart50TokenizerFast.from_pretrained(self.model_name, src_lang=src_lang)
|