peterhung commited on
Commit
6207e35
1 Parent(s): c3e1058

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -41,7 +41,7 @@ import torch
41
  import numpy as np
42
 
43
  def load_trained_transformer_model():
44
- model_path = "peterhung/transformer-vnaccent-marker"
45
  tokenizer = AutoTokenizer.from_pretrained(model_path, add_prefix_space=True)
46
  model = AutoModelForTokenClassification.from_pretrained(model_path)
47
  return model, tokenizer
 
41
  import numpy as np
42
 
43
  def load_trained_transformer_model():
44
+ model_path = "peterhung/vietnamese-accent-marker-xlm-roberta"
45
  tokenizer = AutoTokenizer.from_pretrained(model_path, add_prefix_space=True)
46
  model = AutoModelForTokenClassification.from_pretrained(model_path)
47
  return model, tokenizer