Downloads · 30 days
10
1% of all-time downloads
Zayt/viRoberta-l6-h384-word-cased
viRoberta-l6-h384-word-cased is a fill-mask model from Zayt. Use it when you need the model to fill a missing word. It is set up for transformers.
Downloads · 30 days
10
1% of all-time downloads
All-time downloads
1.1K
Public
Repo size
271 MB
Likes
0
Public
Click a slice to open those files.
.bin90.2 MB · 98%
From the Hugging Face model README
More information: github
from underthesea import word_tokenize
from transformers import RobertaTokenizer, RobertaModel
model_name = 'Zayt/viRoberta-l6-h384-word-cased'
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForMaskedLM.from_pretrained(model_name)
text = word_tokenize("Xin chào, tôi không còn là sinh viên đại học Bách Khoa.", format='text')
output = model(**tokenizer(text, return_tensors='pt))
output