Update app.py

app.py

@@ -16,6 +16,8 @@ class BasicAgent:
     def __init__(self):
         self.model = HfApiModel(max_tokens=2096, temperature=0.5, model_id='Qwen/Qwen2.5-Coder-32B-Instruct', custom_role_conversions=None,)
         # CodeAgent allows the model to use a python interpreter to solve complex problems
+        with open("prompts.yaml", 'r') as stream:
+            self.prompt_templates = yaml.safe_load(stream)
         self.agent = CodeAgent(
             model=self.model,
             tools=[], ## add your tools here (don't remove final answer)
@@ -25,7 +27,7 @@ class BasicAgent:
             planning_interval=None,
             name=None,
             description=None,
-            prompt_templates=prompt_templates
+            prompt_templates=self.prompt_templates
         )
         print("Smolagents CodeAgent initialized.")