pgurazada1 commited on
Commit
9690a07
1 Parent(s): 0efc604

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -8,20 +8,20 @@ from langchain_openai import AzureChatOpenAI, ChatOpenAI
8
 
9
  from sklearn.datasets import fetch_openml
10
 
11
- gpt35 = ChatOpenAI(
12
- openai_api_key=os.environ["OPENAI_API_KEY"],
13
- model="gpt-3.5-turbo",
14
- temperature=0
15
- )
16
-
17
- #gpt35 = AzureChatOpenAI(
18
- # api_key=os.environ["AZURE_OPENAI_KEY"],
19
- # azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
20
- # api_version="2023-05-15",
21
- # deployment_name="gpt-35-turbo",
22
  # temperature=0
23
  #)
24
 
 
 
 
 
 
 
 
 
25
  bank_data, _ = fetch_openml(data_id=43718, return_X_y=True, parser="auto")
26
 
27
  pandas_agent = create_pandas_dataframe_agent(
 
8
 
9
  from sklearn.datasets import fetch_openml
10
 
11
+ #gpt35 = ChatOpenAI(
12
+ # openai_api_key=os.environ["OPENAI_API_KEY"],
13
+ # model="gpt-3.5-turbo",
 
 
 
 
 
 
 
 
14
  # temperature=0
15
  #)
16
 
17
+ gpt35 = AzureChatOpenAI(
18
+ api_key=os.environ["AZURE_OPENAI_KEY"],
19
+ azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
20
+ api_version="2023-05-15",
21
+ deployment_name="gpt-35-turbo",
22
+ temperature=0
23
+ )
24
+
25
  bank_data, _ = fetch_openml(data_id=43718, return_X_y=True, parser="auto")
26
 
27
  pandas_agent = create_pandas_dataframe_agent(