Update app.py

app.py

@@ -12,13 +12,13 @@ model_name = "SamanthaStorm/abuse-pattern-detector-v2"
 model = RobertaForSequenceClassification.from_pretrained(model_name, trust_remote_code=True)
 tokenizer = RobertaTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
-# Define labels (17 total)
+# Define labels (18 total)
 LABELS = [
     "gaslighting", "mockery", "dismissiveness", "control",
     "guilt_tripping", "apology_baiting", "blame_shifting", "projection",
     "contradictory_statements", "manipulation", "deflection", "insults",
-    "obscure_formal", "recovery_phase", "suicidal_threat", "physical_threat",
-    "extreme_control", "non_abusive"
+    "obscure_formal", "recovery_phase", "non_abusive", "suicidal_threat", "physical_threat",
+    "extreme_control"
 ]
 
 # Custom thresholds for each label (make sure these match your original settings)
@@ -44,7 +44,7 @@ THRESHOLDS = {
 
 }
 
-# Define label groups using slicing (first 14: abuse patterns, last 3: danger cues)
+# Define label groups using slicing (first 15: abuse patterns, last 3: danger cues)
 PATTERN_LABELS = LABELS[:15]
 DANGER_LABELS = LABELS[15:18]