Update app.py

app.py

@@ -11,7 +11,7 @@ import pandas as pd
 import torch
 from functools import lru_cache
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
-from smolagents import CodeAgent, WebSearchTool, WikipediaSearchTool, VisitWebpageTool, PythonInterpreterTool
+from smolagents import PromptTemplates, CodeAgent, WebSearchTool, WikipediaSearchTool, VisitWebpageTool, PythonInterpreterTool
 import smolagents.tools as _tools
 from smolagents.models import ChatMessage
 # from huggingface_hub import InferenceClient, hf_hub_download
@@ -137,14 +137,20 @@ class BasicAgent:
         search_tool = CachedWebSearchTool()
         python_tool = PreloadedPythonTool()
         html_parse_tool = VisitWebpageTool()
+        # System prompt
+        my_templates = PromptTemplates(
+            system_prompt=(
+                "When writing Python code in the PythonInterpreterTool," 
+                "you must always import bs4 and regex (already preloaded)," 
+                "and *return* your result by calling final_answer(...). Do not assign to a variable named final_answer."
+            )
+        )
         # Initialize the agent
         self.agent = CodeAgent(model=self, 
                                tools=[wiki_tool, search_tool, python_tool, html_parse_tool], 
                                add_base_tools=True,
-                               additional_authorized_imports=["dateparser", "bs4", "regex"],
-                               system_message=("When writing Python code in the PythonInterpreterTool, you must "
-                                 "always import bs4 and regex (already preloaded), and *return* your "
-                                 "result by calling final_answer(...). Do not assign to a variable named final_answer.")
+                               prompt_templates=my_templates, 
+                               additional_authorized_imports=["dateparser", "bs4", "regex"]
          )
         
     def _serialize_messages(self, messages):