Views
No views yet
cointegrated/rubert-tiny2| Accuracy | Precision | Recall | F1-score |
|---|---|---|---|
| 0.996342 | 0.999747 | 0.993717 | 0.996723 |
1from huggingface_hub import hf_hub_download
2from transformers import AutoTokenizer, AutoModelForSequenceClassification
3import torch
4import pickle
5
6device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
7model_path = 'desllre/ru_news_detection'
8
9encoder_path = hf_hub_download(repo_id=model_path, filename="encoder.pkl")
10with open(encoder_path, "rb") as f:
11 encoder = pickle.load(f)
12
13tokenizer = AutoTokenizer.from_pretrained(model_path)
14classifier = AutoModelForSequenceClassification.from_pretrained(model_path).to(device)
15
16text = 'Tesla дала добро на взлом ПО своих автомобилей\n\nКомпания изменила условия программы Bug Bounty, предусматривающей выплату вознаграждений за поиск уязвимостей. Теперь энтузиасты могут взламывать электрокары Tesla, не боясь отзыва гарантии. Более того, в соответствии с новой политикой компании, автопроизводитель будет перепрошивать автомобили, ПО которых вышло из строя в процессе экспериментов специалистов кибербезопасности.\n\nИзменения в политике компании Telsa очень тепло встретили представители индустрии.'
17
18tokenized = tokenize_function(text, news_tokenizer)
19tokenized = {key: value.to(device) for key, value in tokenized.items()}
20with torch.no_grad():
21 output = classifier(**tokenized)
22
23predicted_class_id = torch.argmax(output.logits, dim=1).item()
24label = encoder.inverse_transform([predicted_class_id])[0]
25
26print(label)