Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4# Load model and tokenizer
5model_name = "Abuzaid01/Ai_Human_text_detect"
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
9# Prepare text for classification
10text = "Your text to classify goes here."
11inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512, padding=True)
12
13# Run inference
14with torch.no_grad():
15 outputs = model(**inputs)
16 logits = outputs.logits
17
18# Get the predicted class and probabilities
19probabilities = torch.nn.functional.softmax(logits, dim=1)
20predicted_class_idx = torch.argmax(probabilities, dim=1).item()
21confidence = probabilities[0][predicted_class_idx].item()
22
23# Map class index to label
24labels = ["Human-written", "AI-generated"]
25predicted_label = labels[predicted_class_idx]
26
27print(f"Prediction: {predicted_label}")
28print(f"Confidence: {confidence:.4f}")