Update app.py

app.py

@@ -21,13 +21,20 @@ def ask_polite_guard(input_text:str)-> str:
     """
 
 try:
+     
+    classifier = pipeline("text-classification", "Intel/polite-guard")
+
+    # result = polite_guard.check_politeness(user_input)
+    #if result.is_polite:
+    #    return "Thank you for your polite message! How can I assist you?"
+    #else:
+    #    return "Please rephrase your message to be more polite."
+
+
+    return classifier(input_text)
     
-    if input_text: 
-        classifier = pipeline("text-classification", "Intel/polite-guard")
-        text = "str"
-        return classifier(input_text)
  except Exception as e:
-    return f"Error fetching classification for text '{input_text}': {str(e)}"
+        return f"Error fetching classification for text '{input_text}': {str(e)}"
 
 
 @tool