datasetsANDmodels commited on
Commit
846c95f
·
verified ·
1 Parent(s): 486a17b

Rename m2m.py to usage.py

Browse files
Files changed (1) hide show
  1. m2m.py → usage.py +2 -2
m2m.py → usage.py RENAMED
@@ -4,8 +4,8 @@ import time
4
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
5
  sr= sys.argv[1]
6
  tr=sys.argv[2]
7
- model = M2M100ForConditionalGeneration.from_pretrained("m2m")
8
- tokenizer = M2M100Tokenizer.from_pretrained("m2m")
9
  tokenizer.src_lang =sr
10
  text="I need to buy a tablet for my mum as soon as possible"
11
  #text="Tôi cần mua một viên thuốc cho mẹ tôi càng sớm càng tốt" VIET
 
4
  from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
5
  sr= sys.argv[1]
6
  tr=sys.argv[2]
7
+ model = M2M100ForConditionalGeneration.from_pretrained("m2m-translator")
8
+ tokenizer = M2M100Tokenizer.from_pretrained("m2m-translator")
9
  tokenizer.src_lang =sr
10
  text="I need to buy a tablet for my mum as soon as possible"
11
  #text="Tôi cần mua một viên thuốc cho mẹ tôi càng sớm càng tốt" VIET