Update README.md

README.md

@@ -26,8 +26,8 @@ This is the base version of our model. See also: [base_deep](hlekkur), [big](hle
 
 ```python
 from fairseq.models.transformer import TransformerModel
-TRANSLATION_MODEL_PATH = 'path/to/model.pt'
-TRANSLATION_MODEL = TransformerModel.from_pretrained('path/to/path', checkpoint_file=TRANSLATION_MODEL_PATH, bpe='sentencepiece', sentencepiece_model='sentencepiece.bpe.model')
+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)