Views
No views yet
eval_metrics.json.1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tok = AutoTokenizer.from_pretrained("Dc-4nderson/TheCulture-content-moderation-model")
5model = AutoModelForSequenceClassification.from_pretrained("Dc-4nderson/TheCulture-content-moderation-model")
6
7inputs = tok("DM me for exclusive deals not available here", return_tensors="pt")
8with torch.no_grad():
9 logits = model(**inputs).logits
10pred = logits.argmax(-1).item()