datacipen commited on
Commit
1f31e36
·
verified ·
1 Parent(s): e6d98f4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -53,7 +53,7 @@ def create_agent(filename: str):
53
  # Create an OpenAI object.
54
  os.environ['OPENAI_API_KEY'] = os.environ['OPENAI_API_KEY']
55
  #llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
56
- llm = OpenAI(temperature=0, model_name="gpt-4-turbo-2024-04-09")
57
  # Read the CSV file into a Pandas DataFrame.
58
  df = pd.read_csv(filename)
59
 
 
53
  # Create an OpenAI object.
54
  os.environ['OPENAI_API_KEY'] = os.environ['OPENAI_API_KEY']
55
  #llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
56
+ llm = OpenAI(temperature=0, model_name="gpt-4")
57
  # Read the CSV file into a Pandas DataFrame.
58
  df = pd.read_csv(filename)
59