Update README.md
Browse files
README.md
CHANGED
@@ -38,7 +38,6 @@ tagger_model_name = "IlyaGusev/rubertconv_toxic_editor"
|
|
38 |
|
39 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
40 |
device_num = 0 if device == "cuda" else -1
|
41 |
-
tagger_tokenizer = AutoTokenizer.from_pretrained(tagger_model_name)
|
42 |
tagger_pipe = pipeline(
|
43 |
"token-classification",
|
44 |
model=tagger_model_name,
|
|
|
38 |
|
39 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
40 |
device_num = 0 if device == "cuda" else -1
|
|
|
41 |
tagger_pipe = pipeline(
|
42 |
"token-classification",
|
43 |
model=tagger_model_name,
|