TiberiuCristianLeon commited on
Commit
a2c2946
·
verified ·
1 Parent(s): de649ff

Update src/Translate.py

Browse files
Files changed (1) hide show
  1. 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.sl}_{self.sl.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)
 
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)