Update tools/polite_guard.py

tools/polite_guard.py

@@ -42,7 +42,11 @@ class PoliteGuardTool(Tool):
             tuple: with classification label and the score 
         """
         try:
-            classifier = pipeline("text-classification", "Intel/polite-guard")
+            classifier = pipeline("text-classification", 
+                                  "Intel/polite-guard", 
+                                   return_all_scores=True,  # <-- get all four probabilities in one call
+                                   truncation=True
+            )
             result = classifier(input_text)
             print(f"return {str(result)}")
             output = result[0]