Update app.py

app.py

@@ -9,18 +9,24 @@ from Gradio_UI import GradioUI
 
 
 @tool
-def get_geolocation(city:str)-> str: 
+"""def get_geolocation(city:str)-> str: 
     """A tool that fetches the geo location of the city (e.g., "latitude":40.71427,"longitude":-74.00597) 
     Args:
         city: A string representing a valid city (e.g., New York)
     """
-    api_url = f"https://geocoding-api.open-meteo.com/v1/search?name={city}&count=1&language=en&format=json"
-    response = requests.get(api_url)
-    if response.status_code == 200:
-        data = response.json()
-        return data.get("latitude", "longitude")
-    else:
-        return "Error: Unable to fetch geo location."
+    Try:"""
+
+@tool
+def model_download_tool(task: str) -> str:
+    """This is a tool that returns the most downloaded model of a given task on the Hugging Face Hub.
+    It returns the name of the checkpoint.
+    
+    Args:
+        task: The task for which to get the download count.
+    """
+    most_downloaded_model = next(iter(list_models(filter=task, sort="download", direction=-1)))
+
+    return most_downloaded_model.id
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -60,7 +66,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,get_geolocation,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
+    tools=[final_answer,most_downloaded_model], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,