Update app.py

app.py

@@ -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['Product Name'].str.contains(arg1)]
+    responses = dataset[dataset['train']['Product Name'].str.contains(arg1)]
     print(responses)
     #filter_dataset_min = filter_dataset['Selling Price'].min()
     #filter_dataset_max = filter_dataset['Selling Price'].min()