Update app.py

app.py

@@ -4,7 +4,7 @@ from smolagents import tool
 #from smolagents import DuckDuckGoSearchTool
 import os
 from datasets import load_dataset
-dataset = load_dataset("bprateek/amazon_product_description", revision="main", token=os.getenv('Testing'))
+dataset = load_dataset("bprateek/amazon_product_description", revision="main", split="train," token=os.getenv('Testing'))
 
 
 @tool
@@ -16,7 +16,7 @@ def get_price_tool(arg1:str)-> float: #it's import to specify the return type
     """
     #filter_dataset = set(dataset['train']['Category'])
     print(arg1)
-    responses = dataset[dataset['train']['Product Name'].str.contains(arg1)]
+    responses = dataset[dataset['Product Name'].str.contains(arg1)]
     print(responses)
     #filter_dataset_min = filter_dataset['Selling Price'].min()
     #filter_dataset_max = filter_dataset['Selling Price'].min()