Views
No views yet
0 → Safe ✅1 → Blocked 🚫1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("jolmiano91/siteblocker-bert-finetuned")
5model = AutoModelForSequenceClassification.from_pretrained("jolmiano91/siteblocker-bert-finetuned")
6
7inputs = tokenizer("This is a safe educational site.", return_tensors="pt")
8outputs = model(**inputs)
9print(torch.argmax(outputs.logits)) # 0 → Safe ✅