Update main.py
Browse files
main.py
CHANGED
@@ -58,7 +58,7 @@ def create_agent(filename: str):
|
|
58 |
df = pd.read_csv(filename)
|
59 |
|
60 |
# Create a Pandas DataFrame agent.
|
61 |
-
return create_pandas_dataframe_agent(llm, df, verbose=False)
|
62 |
|
63 |
def query_agent(agent, query):
|
64 |
"""
|
|
|
58 |
df = pd.read_csv(filename)
|
59 |
|
60 |
# Create a Pandas DataFrame agent.
|
61 |
+
return create_pandas_dataframe_agent(llm, df, verbose=False, allow_dangerous_code=True)
|
62 |
|
63 |
def query_agent(agent, query):
|
64 |
"""
|