Update README.md
Browse files
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/
|
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
|