Update app.py

app.py

@@ -67,30 +67,16 @@ model      = AutoModelForSequenceClassification.from_pretrained(model_name)
 tokenizer  = AutoTokenizer.from_pretrained(model_name, use_fast=False)
 
 LABELS = [
-    "blame shifting",
-    "contradictory statements",
-    "control",
-    "dismissiveness",
-    "gaslighting",
-    "guilt tripping",
-    "insults",
-    "obscure language",
-    "projection",
-    "recovery phase",
-    "threat"
-    "deflection"
-    "aggression"
-    "coercion"
-
+    "recovery", "control", "gaslighting", "dismissiveness", "blame shifting",
+    "coercion", "aggression", "nonabusive", "deflection", "projection", "insults"
 ]
 
 THRESHOLDS = {
-    "recovery phase": 0.622,
+    "recovery": 0.622,
     "control": 0.100,
     "gaslighting": 0.410,
     "dismissiveness": 0.867,
     "blame shifting": 0.116,
-    "contradictory statements": 0.100,
     "coercion": 0.100,
     "aggression": 0.100,
     "nonabusive": 0.100,
@@ -98,17 +84,19 @@ THRESHOLDS = {
     "projection": 0.100,
     "insults": 0.100
 }
+
 PATTERN_WEIGHTS = {
-    "gaslighting": 1.4,       # Slightly reduced: threshold tuned to 0.41 (high precision)
-    "control": 1.3,           # Increased: very low threshold, deserves slightly higher weight
-    "dismissiveness": 0.9,    # Increased from 0.7 due to high threshold needed
-    "blame shifting": 0.6,    # Slight increase for stronger detection
-    "guilt tripping": 1.2,    # Leave as is (not in current threshold set, assumed stable)
-    "insults": 1.5,           # Raised slightly: F1 of 1.0 at low threshold, so still high signal
-    "projection": 1.2,        # Leave as is: still valid at current detection levels
-    "recovery phase": 1.0,    # Slightly lowered to reduce false sense of safety
-    "contradictory statements": 0.75,  # Leave as is unless retrained
-    "threat": 1.6             # Keep high due to high severity, even if threshold wasn't tuned here
+    "gaslighting": 1.5,
+    "control": 1.2,
+    "dismissiveness": 0.7,
+    "blame shifting": 0.5,
+    "insults": 1.4,
+    "projection": 1.2,
+    "recovery": 1.1,
+    "coercion": 1.3,
+    "aggression": 1.4,
+    "nonabusive": 0.1,
+    "deflection": 0.4
 }
 RISK_STAGE_LABELS = {
     1: "🌀 Risk Stage: Tension-Building\nThis message reflects rising emotional pressure or subtle control attempts.",