File size: 939 Bytes
dfdd24c 2d7fffc dfdd24c 2d7fffc dfdd24c 2d7fffc b44e211 2d7fffc b44e211 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
---
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="bae83dd"
)
model = AutoModelForTokenClassification.from_pretrained(model_name)
pipe = pipeline(
"token-classification",
model=model,
tokenizer=tokenizer,
device=-1,
aggregation_strategy="none",
ignore_labels=("NO",)
)
text = "щеколда"
print(text)
index = pipe(text)[0]["index"]
print(text[:index] + "'" + text[index:])
```
Colab: [link](https://colab.research.google.com/drive/1I61aDezhxMVZzHQQfpn7Wqn-ydbndO6i) |