Update tools/polite_guard.py

tools/polite_guard.py

@@ -45,11 +45,7 @@ class PoliteGuardTool(Tool):
             classifier = pipeline("text-classification", "Intel/polite-guard")
             result = classifier(input_text)
             print(f"return {str(result)}")
-            print(f"length is {len(result)}")
-            # get the first tuple 
-            label = result[0]['label']
-            score = result[0]['score']
-            return label, score
+            return result[0]
             
         except Exception as e:
             return f"Error fetching classification for text '{input_text}': {str(e)}"