Update app.py

app.py

@@ -14,8 +14,9 @@ def predict_price_tool(arg1:str)-> float: #it's import to specify the return typ
     Args:
         arg1: the category of product
     """
-    filter_dataset = set(dataset['train']['Category'])
-    print(filter_dataset)
+    #filter_dataset = set(dataset['train']['Category'])
+    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()
     return 1
@@ -25,7 +26,7 @@ def predict_price_tool(arg1:str)-> float: #it's import to specify the return typ
 #Agent Example 
 model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=os.getenv('Testing'))
 agent = CodeAgent(tools=[predict_price_tool], model=model)
-agent.run("Get price quoatition for catageory = Toys & Games | Arts & Crafts | Craft Kits | Paper Craft")
+agent.run("Electronics")
 
 # Access HF Hub 
 #from huggingface_hub import list_models