Update app.py

app.py

@@ -11,26 +11,6 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
-@tool
-def ask_polite_guard(input_text:str)-> str:
-    """ Tool that classifies text into four categories: polite, somewhat polite, neutral and impolite
-    
-    Args:
-        arg1: the first argument
-        arg2: the second argument
-    """
-    try:
-        classifier = pipeline("text-classification", "Intel/polite-guard")
-        return classifier(input_text)
-    except Exception as e:
-        return f"Error fetching classification for text '{input_text}': {str(e)}"
-        # 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."
-        
-
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -74,7 +54,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, ask_polite_guard, get_current_time_in_timezone ], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone ], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,