hungary-mszp-dk-villain

from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

model_name = "Politics/hungary-mszp-dk-villain/"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

text = "A dollárbaloldal ismét elárulta a hazát Brüsszelben."

inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
with torch.no_grad():
    logits = model(**inputs).logits

predicted_class_id = logits.argmax().item()
print(f"Prediction: {predicted_class_id}")
# Output: 1 (Villain) or 0 (Neutral)