Update README.md

README.md

@@ -28,10 +28,30 @@ Use in pair with [mask filler](https://huggingface.co/IlyaGusev/sber_rut5_filler
 
 #### How to use
 
-Colab: [link](TBA)
+Colab: [link](https://colab.research.google.com/drive/1NUSO1QGlDgD-IWXa2SpeND089eVxrCJW)
 
 ```python
-TBA
+import torch
+from transformers import AutoTokenizer, pipeline
+
+tagger_model_name = "IlyaGusev/rubertconv_toxic_editor"
+
+device = "cuda" if torch.cuda.is_available() else "cpu"
+device_num = 0 if device == "cuda" else -1
+tagger_tokenizer = AutoTokenizer.from_pretrained(tagger_model_name)
+tagger_pipe = pipeline(
+    "token-classification",
+    model=tagger_model_name,
+    tokenizer=tagger_model_name,
+    framework="pt",
+    device=device_num,
+    aggregation_strategy="max"
+)
+
+text = "..."
+tagger_predictions = tagger_pipe([text], batch_size=1)
+sample_predictions = tagger_predictions[0]
+print(sample_predictions)
 ```
 
 ## Training data