Update app.py

app.py

@@ -13,6 +13,7 @@ from smolagents import (
     Tool,
     OpenAIServerModel,
     SpeechToTextTool,
+    FinalAnswerTool,
 )
 
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -59,12 +60,13 @@ class LoadTextFileTool(Tool):
 prompts = yaml.safe_load(
     importlib.resources.files("smolagents.prompts").joinpath("code_agent.yaml").read_text()
 )
-prompts["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. "
+prompts["system_prompt"] = ("You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer using the FinalAnswerTool. Your final answer MUST 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. "
                             + prompts["system_prompt"])
+print(prompts["system_prompt"])
 
 def init_agent():
     model = OpenAIServerModel(
-            model_id="gpt-4.1-mini",
+            model_id="gpt-4.1",
             api_key=os.getenv("OPENAI_API_KEY"),
             temperature=0.7
     )
@@ -77,6 +79,7 @@ def init_agent():
             SpeechToTextTool(),
             LoadXlsxFileTool(),
             LoadTextFileTool(),
+            FinalAnswerTool(),
         ],
         model=model,
         prompt_templates=prompts,