saaketvarma commited on
Commit
9623319
·
1 Parent(s): c8d18cb

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -12,7 +12,7 @@ def query_agent(data, query):
12
 
13
  llm = OpenAI()
14
  Set the model to use
15
- model = "gpt-4-32k"
16
 
17
  # Create a Pandas DataFrame agent.
18
  agent = create_pandas_dataframe_agent(llm, df, model= model, verbose=True)
 
12
 
13
  llm = OpenAI()
14
  Set the model to use
15
+ model = "gpt4"
16
 
17
  # Create a Pandas DataFrame agent.
18
  agent = create_pandas_dataframe_agent(llm, df, model= model, verbose=True)