Update app.py

app.py

@@ -23,17 +23,16 @@ custom_role_conversions=None,
 )
 
 @tool
-def f1_tackinfo_getter(track_name: 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
+def f1_tackinfo_getter(country: str)-> str: #it's import to specify the return type
     """
     Returns data for a specified race
     Args:
-        track_name: A string respresenting a valid track name from the 2024 F1 calendar
+        country: A string respresenting a valid country name from the 2024 F1 calendar.
     Returns:
         A string with information about the given race
     """
     df = pd.read_csv('./Formula1_2024season_raceResults.csv')
-    info =  str(df.groupby('Track').get_group(track_name))
+    info =  str(df.groupby('Track').get_group(country))
     client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
     system_prompt = "You are an expert in F1 race analysis. You will be given data about a race and your goal is to provide a concise analysis of these recults"
     output = client.chat.completions.create(