Update app.py

app.py

@@ -10,12 +10,12 @@ from typing import List, Dict, Any
 
 # Define categories with sub-level information
 CATEGORIES = {
-    'hateful': ['level_1_discriminatory', 'level_2_hate_speech'],
+    'hateful': ['hateful_lvl_1_discriminatory', 'hateful_lvl_2_hate_speech'],
     'insults': ['insults'],
-    'sexual': ['level_1_not_appropriate_for_minors', 'level_2_not_appropriate_for_all_ages'],
+    'sexual': ['sexual_lvl_1_not_appropriate_for_minors', 'sexual_lvl_2_not_appropriate_for_all'],
     'physical_violence': ['physical_violence'],
-    'self_harm': ['level_1_self_harm_intent', 'level_2_self_harm_action'],
-    'all_other_misconduct': ['level_1_not_socially_accepted', 'level_2_illegal_activities']
+    'self_harm': ['self_harm_lvl_1_intent', 'self_harm_lvl_2_action'],
+    'all_other_misconduct': ['all_other_misconduct_lvl_1_not_socially_accepted', 'all_other_misconduct_lvl_2_illegal']
 }
 
 def get_embeddings(texts: List[str], model: str = "text-embedding-3-large") -> np.ndarray: