Update app.py

app.py

@@ -7,9 +7,9 @@ import subprocess
 import pandas as pd
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-from smolagents import CodeAgent, HfApiModel
+from smolagents import CodeAgent, WebSearchTool, WikipediaSearchTool
 from smolagents.models import ChatMessage
-from huggingface_hub import InferenceClient, hf_hub_download
+# from huggingface_hub import InferenceClient, hf_hub_download
 
 subprocess.run(["playwright", "install"], check=True)
 
@@ -85,7 +85,7 @@ class BasicAgent:
         # Initialize the tools other than the base tools
         # See list of base tools in https://github.com/huggingface/smolagents/blob/main/src/smolagents/default_tools.py
         
-        # Initialize the agent
+        # Download the model weights and build the pipeline
         tok = AutoTokenizer.from_pretrained(model_id, token=hf_token)
         mod = AutoModelForCausalLM.from_pretrained(
             model_id,
@@ -101,7 +101,14 @@ class BasicAgent:
             return_full_text=False,      # <— only get the completion, not the prompt + completion
             # temperature=1.0,
         )
-        self.agent = CodeAgent(model=self, tools=[], add_base_tools=True)
+        # Introduce tools
+        wiki_tool = WikipediaSearchTool()
+        search_tool = WebSearchTool()
+        # Initialize the agent
+        self.agent = CodeAgent(model=self, 
+                               tools=[wiki_tool, search_tool], 
+                               add_base_tools=True,
+                               additional_authorized_imports=["dateparser"])
         
     def _serialize_messages(self, messages):
         prompt = []