arnastofnun
/

wmt24-en-is-transformer-big

Model card Files Files and versions Community

atlijas commited on Aug 26

Commit

9940781

•

1 Parent(s): 3d52e03

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -29,7 +29,7 @@ from fairseq.models.transformer import TransformerModel
 TRANSLATION_MODEL_NAME = 'checkpoint_best.pt'
 TRANSLATION_MODEL = TransformerModel.from_pretrained('path/to/model', checkpoint_file=TRANSLATION_MODEL_NAME, bpe='sentencepiece', sentencepiece_model='sentencepiece.bpe.model')
 src_sentences = ['This is a test sentence.', 'This is another test sentence.']
-translated_sentences = translate(translation_model=TRANSLATION_MODEL, sentences=src_sentences, beam=5)
 print(translated_sentences)
 ```

 TRANSLATION_MODEL_NAME = 'checkpoint_best.pt'
 TRANSLATION_MODEL = TransformerModel.from_pretrained('path/to/model', checkpoint_file=TRANSLATION_MODEL_NAME, bpe='sentencepiece', sentencepiece_model='sentencepiece.bpe.model')
 src_sentences = ['This is a test sentence.', 'This is another test sentence.']
+translated_sentences = TRANSLATION_MODEL.translate(src_sentences)
 print(translated_sentences)
 ```