Update app.py

app.py

@@ -3,7 +3,7 @@ from smolagents import HfApiModel
 from smolagents import tool
 #from smolagents import DuckDuckGoSearchTool
 import os
-from Gradio_UI import GradioUI
+import gradio as gr
 from datasets import load_dataset
 dataset = load_dataset("bprateek/amazon_product_description", revision="main", split="train", token=os.getenv('Testing'))
 
@@ -24,9 +24,12 @@ 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'))
-agent = CodeAgent(tools=[get_price_tool], model=model)
+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")
-GradioUI(agent).launch()
 
 # Access HF Hub 
 #from huggingface_hub import list_models