Update app.py

app.py

@@ -88,13 +88,13 @@ def run_model(model_file: str, embeddings: np.ndarray):
     expanded_probabilities = []
     expanded_label_names = []
     for i, cat in enumerate(label_names):
-        # Level 1: binary indicator for class 1 only.
-        y_pred_l1 = (predictions[:, i] == 1).astype(int)
-        y_proba_l1 = probabilities[:, i, 1]  # Probability for class 1
+        # Level 1 binary 
+        y_pred_l1 = (predictions[:, i] > 0).astype(int) # y == 1 or y == 2
+        y_proba_l1 = 1 - probabilities[:, i, 0]  # 1 - P(class 0)
         
-        # Level 2: binary indicator for any positivity (class 1 or 2)
-        y_pred_l2 = (predictions[:, i] > 0).astype(int)
-        y_proba_l2 = 1 - probabilities[:, i, 0]  # 1 - probability for class 0
+        # Level 2 binary 
+        y_pred_l2 = (predictions[:, i] == 2).astype(int) # only y == 2
+        y_proba_l2 = probabilities[:, i, 2]  # Probability of class 2
         
         # For certain categories, only the overall (level 2) output is used
         if cat in ['binary', 'insults', 'physical_violence']: