samim2024 commited on
Commit
4900924
·
verified ·
1 Parent(s): 115765e

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -13,7 +13,7 @@ def query_agent(data, query):
13
  df = pd.read_csv(data)
14
 
15
  #llm = OpenAI()
16
- llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-Instruct-v0.2", temperature=0.9)
17
  # Create a Pandas DataFrame agent.
18
  agent = create_pandas_dataframe_agent(llm, df, verbose=True)
19
 
 
13
  df = pd.read_csv(data)
14
 
15
  #llm = OpenAI()
16
+ llm = HuggingFaceEndpoint(repo_id="mistralai/Mixtral-8x7B-Instruct-v0.1", temperature=0.9)
17
  # Create a Pandas DataFrame agent.
18
  agent = create_pandas_dataframe_agent(llm, df, verbose=True)
19