testing

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
 import datetime
 import requests
 import pytz
@@ -9,14 +9,15 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_cutom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return type
+def eat_pizza(arg1:str, arg2:int)-> str: #it's import to specify the return type
     #Keep this format for the description / args / args description but feel free to modify the tool
     """A tool that does nothing yet 
     Args:
         arg1: the first argument
         arg2: the second argument
     """
-    return f"What magic will you build with {arg1} and {arg2} ?"
+    return f"You ate {arg2} {arg1}!"
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -42,16 +43,17 @@ model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',#
 custom_role_conversions=None,
 )
 
-
 # Import tool from Hub
 image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
+# "Qwen/Qwen2.5-72B-Instruct"
+
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, DuckDuckGoSearchTool, get_current_time_in_timezone, my_cutom_tool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, DuckDuckGoSearchTool(), image_generation_tool, get_current_time_in_timezone, eat_pizza],
     max_steps=6,
     verbosity_level=1,
     grammar=None,