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

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -11,8 +11,8 @@ def query_agent(data, query):
11
  df = pd.read_csv(data)
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)
 
11
  df = pd.read_csv(data)
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)