odegiber commited on
Commit
b1d59bd
·
verified ·
1 Parent(s): f0d7c92

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -31,7 +31,7 @@ The model is loaded and used with `transformers` as:
31
  from transformers import MarianMTModel, MarianTokenizer
32
  import torch
33
 
34
- model_name = "models/hf/eo_esenca_shuf"
35
 
36
  device = "cuda" if torch.cuda.is_available() else "cpu"
37
  model = MarianMTModel.from_pretrained(model_name).to(device)
 
31
  from transformers import MarianMTModel, MarianTokenizer
32
  import torch
33
 
34
+ model_name = "Helsinki-NLP/opus-mt-eo-caenes"
35
 
36
  device = "cuda" if torch.cuda.is_available() else "cpu"
37
  model = MarianMTModel.from_pretrained(model_name).to(device)