Update utils.py

utils.py

@@ -3,15 +3,17 @@ from langchain_experimental.agents import create_pandas_dataframe_agent
 import pandas as pd
 #from langchain.llms import OpenAI
 #New import from langchain, which replaces the above
-from langchain_openai import OpenAI
+#from langchain_openai import OpenAI
+from langchain.llms import HuggingFaceEndpoint
+from langchain.llms import huggingface_hub
 
 def query_agent(data, query):
 
     # Parse the CSV file and create a Pandas DataFrame from its contents.
     df = pd.read_csv(data)
 
-    llm = OpenAI()
-    
+    #llm = OpenAI()
+    llm = HuggingFaceEndpoint(repo_id="mistralai/Mistral-7B-Instruct-v0.2", temperature=0.9)
     # Create a Pandas DataFrame agent.
     agent = create_pandas_dataframe_agent(llm, df, verbose=True)