system prompt and use zephyr

app.py

@@ -5,24 +5,39 @@ import inspect
 import pandas as pd
 
 # Updated smolagents import
-from smolagents import CodeAgent, DuckDuckGoSearchTool
+from smolagents import (
+    CodeAgent,
+    DuckDuckGoSearchTool,
+    HfApiModel,
+)  # Add HfApiModel import
 
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 
+SYSTEM_PROMPT = (
+    "You are a general AI assistant. I will ask you a question. "
+    "Report your thoughts, and finish your answer with the following template: "
+    "FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. "
+    "If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. "
+    "If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. "
+    "If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."
+)
+
+
 class MyAgent:
     def __init__(self):
         print("MyAgent initialized.")
-        # Standard tools for CodeAgent
         self.tools = [DuckDuckGoSearchTool()]
-        # You can add more tools if needed
-        self.agent = CodeAgent(tools=self.tools)
+        # Use the specified model
+        self.model = HfApiModel(repo_id="HuggingFaceH4/zephyr-7b-beta")
+        self.agent = CodeAgent(tools=self.tools, model=self.model)
 
     def __call__(self, question: str) -> str:
-        # Use the agent to answer the question
-        result = self.agent.run(question)
+        # Prepend the system prompt to the question
+        prompt = f"{SYSTEM_PROMPT}\n\nQuestion: {question}"
+        result = self.agent.run(prompt)
         return result