🚧 Some cleaning

app.py

@@ -1,6 +1,5 @@
 from typing import Any
 import os
-import yaml
 import gradio as gr
 import pandas as pd
 import requests
@@ -20,6 +19,7 @@ from smolagents import (
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
+
 final_answer = FinalAnswerTool()
 web_search = WebSearchTool(engine="bing")
 visit_webpage = VisitWebpageTool()
@@ -110,7 +110,14 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
         agent = GaiaAgent(
             api_url,
             model=model,
-            tools=[final_answer, web_search, visit_webpage, wiki_search],
+            tools=[
+                final_answer,
+                web_search,
+                visit_webpage,
+                wiki_search,
+                list_wikipedia_sections,
+                count_tables_in_wikipedia_section,
+            ],
             max_steps=6,
             verbosity_level=1,
             grammar=None,