Update README.md

README.md

@@ -72,7 +72,7 @@ pip install -q ./transformers
 ```python
 from transformers import MBart50TokenizerFast, MBartForConditionalGeneration
 
-ckpt = 'mbart-large-50-finetuned-opus-en-pt-translation'
+ckpt = 'Narrativa/mbart-large-50-finetuned-opus-en-pt-translation'
 
 tokenizer = MBart50TokenizerFast.from_pretrained(ckpt)
 model = MBartForConditionalGeneration.from_pretrained(ckpt).to("cuda")
@@ -80,7 +80,6 @@ model = MBartForConditionalGeneration.from_pretrained(ckpt).to("cuda")
 tokenizer.src_lang = 'en_XX'
 
 def translate(text):
-
     inputs = tokenizer(text, return_tensors='pt')
     input_ids = inputs.input_ids.to('cuda')
     attention_mask = inputs.attention_mask.to('cuda')