Update detector.py

detector.py

@@ -111,7 +111,7 @@ def score_text(text, max_len=512):
         probs = torch.softmax(logits, dim=-1).cpu().numpy()
 
     # Extract AI probability (label=1)
-    ai_prob = float(probs[1])
+    ai_prob = float(probs[0][1])
     return ai_prob
 
 # === Artifact Detection ===