Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 import inspect
 import pandas as pd
 
-from smolagents import CodeAgent,DuckDuckGoSearchTool, InferenceClientModel
+from smolagents import CodeAgent,DuckDuckGoSearchTool, InferenceClientModel, PromptTemplates, EMPTY_PROMPT_TEMPLATES
 from tools.final_answer import FinalAnswerTool
 import yaml
 
@@ -24,7 +24,11 @@ class BasicAgent:
         model = InferenceClientModel()
         
         with open("prompts.yaml", 'r') as stream:
-            prompt_templates = yaml.safe_load(stream)
+            prompts = yaml.safe_load(stream)
+
+        prompt_templates = EMPTY_PROMPT_TEMPLATES
+        prompt_templates.system_prompt = prompts['system_prompt']
+        prompt_templates.final_answer.pre_messages = prompts['final_answer']['pre_messages']
             
         self.agent = CodeAgent(
             model=model,