Update app.py

app.py

@@ -9,7 +9,6 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL)
 
 def detect_ai(text):
-    # Split into rough sentences
     sentences = re.split(r'(?<=[.!?]) +', text)
     results = []
 
@@ -20,17 +19,17 @@ def detect_ai(text):
         with torch.no_grad():
             outputs = model(**inputs)
             probs = torch.softmax(outputs.logits, dim=1)
-        ai_score = float(probs[1])
+        ai_score = float(probs[0][1])   # fix here
         results.append({"sentence": sent, "ai_score": ai_score})
 
-    # Build highlighted HTML
     highlighted = ""
     for r in results:
-        color = f"rgba(255,0,0,{r['ai_score']})"  # more red = more AI-like
+        color = f"rgba(255,0,0,{r['ai_score']})"
         highlighted += f"<span style='background-color:{color}; padding:2px'>{r['sentence']} </span>"
 
     return highlighted, results
 
+
 with gr.Blocks() as demo:
     gr.Markdown("## 🤖 AI Detector (like ZeroGPT)")
     gr.Markdown("Paste your text below. Redder highlights = more AI-like.")