Update app.py

app.py

@@ -95,11 +95,10 @@ def run_model(model_file: str, embeddings: np.ndarray):
         # 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']:
-            expanded_predictions.append(y_pred_l2)
-            expanded_probabilities.append(y_proba_l2)
+            expanded_predictions.append(y_pred_l1)
+            expanded_probabilities.append(y_proba_l1)
             expanded_label_names.append(cat)
         else:
             expanded_predictions.append(y_pred_l1)