Update tools/polite_guard.py

tools/polite_guard.py

@@ -30,12 +30,11 @@ class PoliteGuardTool(Tool):
             input_text: The text to classify.
 
         Returns:
-            dictionary with label and score values
+            string with classification label
         """
         try:
             classifier = pipeline("text-classification", "Intel/polite-guard")
-            
-            return classifier['label']
+            return classifier(input_text)
             
         except Exception as e:
             return f"Error fetching classification for text '{input_text}': {str(e)}"