metadata

language:
  - ru
tags:
  - token-classification
license: apache-2.0
inference: false

RuWordStressTransformer

Model description

Transformer encoder for predicting word stress in Russian.

Intended uses & limitations

How to use

from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline

model_name = "IlyaGusev/ru-word-stress-transformer"
tokenizer = AutoTokenizer.from_pretrained(
    model_name,
    trust_remote_code=True,
    revision="3400828"
)
model = AutoModelForTokenClassification.from_pretrained(model_name)
pipe = pipeline(
    "token-classification",
    model=model,
    tokenizer=tokenizer,
    device=-1,
    aggregation_strategy="none",
    ignore_labels=("NO",)
)
print(pipe("щеколда"))