Update app.py

app.py

@@ -9,15 +9,14 @@ from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
 @tool
-def my_custom_tool(year:int)-> str: #it's import to specify the return type
+def get_weather(city:str)-> 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 fetches the population of USA at a specific year. Do a GET call on this API:
-    API: https://datausa.io/api/data?drilldowns=Nation&measures=Population, and get the population of the year from the response.
+    """A tool that fetches the current weather of a city.
     Args:
-        year: an integer representing a year
+        city: A string representing an existing city (e.g paris)
     """
-    response = requests.get('https://datausa.io/api/data?drilldowns=Nation&measures=Population')
-    return response.json()
+
+    return f"The current weather in {city} is cloudy."
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -43,7 +42,8 @@ final_answer = FinalAnswerTool()
 model = HfApiModel(
 max_tokens=2096,
 temperature=0.5,
-model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
+model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', 
+#model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 custom_role_conversions=None,
 )
 
@@ -56,7 +56,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_weather], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,