IlyaGusev
/

ru-word-stress-transformer

Token Classification

Model card Files Files and versions Community

IlyaGusev commited on Dec 31, 2022

Commit

b44e211

•

1 Parent(s): bae83dd

Update README.md

Files changed (1) hide show

README.md +8 -3

README.md CHANGED Viewed

@@ -25,7 +25,7 @@ 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(
@@ -36,5 +36,10 @@ pipe = pipeline(
     aggregation_strategy="none",
     ignore_labels=("NO",)
 )
-print(pipe("щеколда"))
-```

 tokenizer = AutoTokenizer.from_pretrained(
     model_name,
     trust_remote_code=True,
+    revision="bae83dd"
 )
 model = AutoModelForTokenClassification.from_pretrained(model_name)
 pipe = pipeline(
     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)