Update tools/polite_guard.py

tools/polite_guard.py

@@ -30,15 +30,16 @@ class PoliteGuardTool(Tool):
             input_text: The text to classify.
 
         Returns:
-            dictionary with
+            string with
                 a. A in result['label'] which is one of the four classifications
                 b. A score in result['score']    
         """
         try:
             classifier = pipeline("text-classification", "Intel/polite-guard")
+            
             return {
-                "label": result['label'],
-                "score": result['score']
+                "label": classifier['label'],
+                "score": classifier['score']
             }
             
         except Exception as e: