AndriiPets commited on
Commit
7e83029
1 Parent(s): 9743b50
utils/translators/en_ru/en_ru.py CHANGED
@@ -2,9 +2,9 @@
2
  from transformers import FSMTForConditionalGeneration, FSMTTokenizer
3
  mname = "facebook/wmt19-en-ru"
4
  tokenizer = FSMTTokenizer.from_pretrained(
5
- mname, cache_dir='../../cache')
6
  model = FSMTForConditionalGeneration.from_pretrained(
7
- mname, cache_dir='../../cache')
8
 
9
 
10
  def translator_en_ru(text):
 
2
  from transformers import FSMTForConditionalGeneration, FSMTTokenizer
3
  mname = "facebook/wmt19-en-ru"
4
  tokenizer = FSMTTokenizer.from_pretrained(
5
+ mname, cache_dir='../../../cache')
6
  model = FSMTForConditionalGeneration.from_pretrained(
7
+ mname, cache_dir='../../../cache')
8
 
9
 
10
  def translator_en_ru(text):
utils/translators/en_ru/ru_en.py CHANGED
@@ -2,9 +2,9 @@
2
  from transformers import FSMTForConditionalGeneration, FSMTTokenizer
3
  mname = "facebook/wmt19-ru-en"
4
  tokenizer = FSMTTokenizer.from_pretrained(
5
- mname, cache_dir='../../cache')
6
  model = FSMTForConditionalGeneration.from_pretrained(
7
- mname, cache_dir='../../cache')
8
 
9
 
10
  def translate_ru_en(text):
 
2
  from transformers import FSMTForConditionalGeneration, FSMTTokenizer
3
  mname = "facebook/wmt19-ru-en"
4
  tokenizer = FSMTTokenizer.from_pretrained(
5
+ mname, cache_dir='../../../cache')
6
  model = FSMTForConditionalGeneration.from_pretrained(
7
+ mname, cache_dir='../../../cache')
8
 
9
 
10
  def translate_ru_en(text):