Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4# Load model and tokenizer
5tokenizer = AutoTokenizer.from_pretrained("yaya36095/text-detector")
6model = AutoModelForSequenceClassification.from_pretrained("yaya36095/text-detector")
7
8def detect_text(text):
9 # Tokenize input
10 inputs = tokenizer(text, return_tensors="pt", truncation=True, max_length=512)
11
12 # Get prediction
13 with torch.no_grad():
14 outputs = model(**inputs)
15 predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
16
17 # Process results
18 scores = predictions[0].tolist()
19 results = [
20 {"label": "HUMAN", "score": scores[0]},
21 {"label": "AI", "score": scores[1]}
22 ]
23
24 return {
25 "prediction": results[0]["label"],
26 "confidence": f"{results[0]['score']*100:.2f}%",
27 "detailed_scores": [
28 f"{r['label']}: {r['score']*100:.2f}%" for r in results
29 ]
30 }