Update app.py

app.py

@@ -14,7 +14,7 @@ from Gradio_UI import GradioUI
 
 # Tool to classify text into politeness categories
 @tool
-def ask_polite_guard(input_text: str) -> str:
+def ask_polite_guard(input_text: str) -> dict:
     """Tool that classifies text into four categories: polite, somewhat polite, neutral, and impolite.
 
     Args:
@@ -22,7 +22,11 @@ def ask_polite_guard(input_text: str) -> str:
     """
     try:
         classifier = pipeline("text-classification", "Intel/polite-guard")
-        return classifier(input_text)
+        return {
+            "label": result['label'],
+            "score": result['score']
+        }
+        
     except Exception as e:
         return f"Error fetching classification for text '{input_text}': {str(e)}"