Update app.py

app.py

@@ -71,21 +71,8 @@ def sql_engine(query: str) -> str:
 
 sql_agent = CodeAgent(
     tools=[sql_engine],
-    model=HfApiModel("meta-llama/Meta-Llama-3.1-8B-Instruct"),
-)
-
-managed_sql_agent = ManagedAgent(
-    agent=sql_agent,
-    name="sql search",
-    description="Runs SQL queries. Give it your query as an argument. Also, this agent should provide the actual query it ran.",
-)
-
-manager_agent = CodeAgent(
-    tools=[],
-    model=HfApiModel(model_id = "Qwen/Qwen2.5-Coder-32B-Instruct"),
-    managed_agents=[managed_sql_agent],
-    additional_authorized_imports=['pyparsing', 'matplotlib', 'datetime', 'statistics', 'bs4', 'request', 'unicodedata', 'queue', 'time', 'collections', 're', 'math', 'stat', 'random', 
-'itertools'],
+    model=HfApiModel(),
+ #   additional_authorized_imports=['pyparsing', 'matplotlib', 'datetime', 'statistics', 'bs4', 'request', 'unicodedata', 'queue', 'time', 'collections', 're', 'math', 'stat', 'random', 'itertools'],
 )
 
 # Function to log agent actions
@@ -97,7 +84,7 @@ def log_agent_action(prompt, result, agent_name):
     st.code(result, language="text")
 
 # Streamlit app title
-st.title("AI SQL Assistant Agent researching your query and summarizing it")
+st.title("AI SQL Agent researching your query and summarizing it")
 
 # App description
 st.write("Generate SQL queries using human speech powered by SmolAgents.")