Update app.py

app.py

@@ -33,6 +33,27 @@ def get_square_root_tool(input_number:int)-> int: #it's import to specify the re
     except Exception as e:
         return f"Error fetching Square root of '{input_number}': {str(e)}" 
 
+@tool
+def classify_educational_article(text: str) -> int:
+    """
+    Classifier for judging the educational value of web pages.
+    Args:
+        article_text: The content of the news article to be classified.
+    Returns:
+        str: This function will output a dictionary with the input text, the predicted score, and an integer score between 0 and 5
+    """
+    from transformers import pipeline
+    # Perform classification
+    model_name = "HuggingFaceTB/fineweb-edu-classifier"
+    classifier = pipeline("text-classification", model=model_name)
+    inputs = tokenizer(text, return_tensors="pt", padding="longest", truncation=True)
+    outputs = classifier(**inputs)
+    logits = outputs.logits.squeeze(-1).float().detach().numpy()
+    score = logits.item()
+    return score
+     
+             
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -70,7 +91,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,image_generation_tool,get_square_root_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer,image_generation_tool,get_square_root_tool,classify_educational_article], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,