nqp426/phobert-ai-detect

PhoBERT AI Text Detection Model

(Mô hình phân loại văn bản: Người viết vs AI viết)

Giới thiệu (Overview)

Mô hình này được fine-tune từ vinai/phobert-base-v2 để phát hiện văn bản tiếng Việt do AI tạo ra.

Mô hình trả về xác suất đoạn văn bản có phải do AI viết hay không.

Sử dụng (Usage)

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)