|
--- |
|
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 |
|
|
|
```python |
|
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("щеколда")) |
|
``` |
