update to Open AI

app.py

@@ -4,18 +4,24 @@ import gradio as gr
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
 
-from langchain_openai import AzureChatOpenAI
+from langchain_openai import AzureChatOpenAI, ChatOpenAI
 
 from sklearn.datasets import fetch_openml
 
-gpt35 = AzureChatOpenAI(
-    api_key=os.environ["AZURE_OPENAI_KEY"],
-    azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
-    api_version="2023-05-15",
-    deployment_name="gpt-35-turbo",
+gpt35 = ChatOpenAI(
+    model_name="gpt-3.5-turbo",
+    api_key=os.environ["OPENAI_API_KEY"],
     temperature=0
 )
 
+#gpt35 = AzureChatOpenAI(
+#    api_key=os.environ["AZURE_OPENAI_KEY"],
+#    azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
+#    api_version="2023-05-15",
+#    deployment_name="gpt-35-turbo",
+#    temperature=0
+#)
+
 bank_data, _ = fetch_openml(data_id=43718, return_X_y=True, parser="auto")
 
 pandas_agent = create_pandas_dataframe_agent(