Update app.py

app.py

@@ -50,15 +50,9 @@ def classify_prompt(text):
             
             # Map model labels to our categories if needed (Arabic)
             label_mapping = {
-                'SAFE': 'آمن',
+                'BENIGN': 'آمن',
                 'INJECTION': 'حقن', 
-                'JAILBREAK': 'كسر حماية',
-                'Safe': 'آمن',
-                'Injection': 'حقن',
-                'Jailbreak': 'كسر حماية',
-                'آمن': 'آمن',
-                'حقن': 'حقن',
-                'كسر حماية': 'كسر حماية'
+                'JAILBREAK': 'كسر حماية'
             }
             
             mapped_label = label_mapping.get(label, label)