from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch

model_id = "nqp426/phobert-ai-detect"

tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForSequenceClassification.from_pretrained(model_id)

text = "Thuật toán học máy ngày càng phát triển mạnh mẽ."
inputs = tokenizer(text, return_tensors="pt", max_length=256, truncation=True)
outputs = model(**inputs)

probs = torch.nn.functional.softmax(outputs.logits, dim=-1)
print(probs)