zerpal-rubert-tiny2 / README.md
codemurt's picture
Update README.md
8562eb1 verified
metadata
library_name: transformers
tags: []

Zerpal-rubert-tiny2

How to use

You can use this model directly with a pipeline for masked language modeling:

from transformers import pipeline

unmasker = pipeline('fill-mask', model='udmurtNLP/zerpal-rubert-tiny2', tokenizer='udmurtNLP/zerpal-rubert-tiny2-tokenizer')

unmasker("Ӟечбур! Мынам нимы [MASK].")

Here is how to use this model to get the features of a given text in PyTorch:

from transformers import AutoTokenizer, AutoModelForMaskedLM
tokenizer = AutoTokenizer.from_pretrained('udmurtNLP/zerpal-rubert-tiny2-tokenizer')
model = AutoModelForMaskedLM.from_pretrained("udmurtNLP/zerpal-rubert-tiny2")
text = "Яратон, яратон, мар меда сыӵе тон?"
encoded_input = tokenizer(text, return_tensors='pt')
output = model(**encoded_input)