Update main.py

main.py

@@ -52,8 +52,8 @@ def create_agent(filename: str):
 
     # Create an OpenAI object.
     os.environ['OPENAI_API_KEY'] = os.environ['OPENAI_API_KEY']
-    #llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
-    llm = OpenAI(temperature=0, model_name="gpt-4")
+    llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
+    #llm = OpenAI(temperature=0, model_name="gpt-4")
     # Read the CSV file into a Pandas DataFrame.
     df = pd.read_csv(filename)