from transformers import AutoTokenizer, AutoModelForSequenceClassification

# Tải mô hình và tokenizer
model_name = "Quintu/roberta-large-768-hazard"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

# Sử dụng mô hình để phân loại văn bản
text = "This is an example text to classify."
inputs = tokenizer(text, return_tensors="pt")
outputs = model(**inputs)

# Dự đoán
logits = outputs.logits
print(logits)