Views
No views yet
ai-forever/ruRoBERTa-largeRuBERTBinaryClassifier0.9989630.9994070.9988151.0000000.55121from huggingface_hub import hf_hub_download
2from transformers import AutoTokenizer
3import torch
4
5
6repo = "DanielNRU/S7-relevance-rubert-20260822"
7tokenizer = AutoTokenizer.from_pretrained(repo)
8model = RuBERTBinaryClassifier("ai-forever/ruRoBERTa-large", hidden_size=512)
9weights = hf_hub_download(repo_id=repo, filename="model.safetensors" if (output_dir / "model.safetensors").exists() else "pytorch_model.bin")
10model.load_state_dict(torch.load(weights, map_location="cpu"))
11model.eval()