Update app.py

app.py

@@ -9,8 +9,7 @@ dataset = load_dataset("bprateek/amazon_product_description", revision="main", s
 
 
 @tool
-def get_price_tool(arg1:str)-> float: #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 get_price_tool(arg1:str)-> float:
     """This is a tool which look on a dataset and filter for products and their various atteibutes like prices description
     Args:
         arg1: the category of product
@@ -24,16 +23,6 @@ def get_price_tool(arg1:str)-> float: #it's import to specify the return type
 
 #Agent Example 
 model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct", token=os.getenv('Testing'))
-demo = gr.Interface(fn=CodeAgent(tools=[get_price_tool], model=model), inputs="textbox", outputs="textbox")
-
-if __name__ == "__main__":
-    demo.launch()
-
-#agent.run("Can you dispay prices of all Electronics products in a table")
-
-# Access HF Hub 
-#from huggingface_hub import list_models
-
-#for model in list_models(limit=10, sort="downloads", direction=-1):
-#   print(model.id, model.downloads)
+agent = CodeAgent(tools=[get_price_tool], model=model)
 
+agent.run("Can you dispay prices of all Electronics products in a table")