Update app.py

app.py

@@ -4,10 +4,18 @@ 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 ChatOpenAI
+from langchain_openai import ChatOpenAI, AzureChatOpenAI
 
 from sklearn.datasets import fetch_openml
 
+gpt4o_azure = AzureChatOpenAI(
+    model_name='gpt-4o-mini',
+    api_key=os.environ["AZURE_OPENAI_KEY"],
+    azure_endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
+    api_version="2024-02-01",
+    temperature=0
+)
+
 gpt35 = ChatOpenAI(
     model_name="gpt-3.5-turbo",
     api_key=os.environ["OPENAI_API_KEY"],
@@ -17,7 +25,7 @@ gpt35 = ChatOpenAI(
 bank_data, _ = fetch_openml(data_id=43718, return_X_y=True, parser="auto")
 
 pandas_agent = create_pandas_dataframe_agent(
-    llm=gpt35,
+    llm=gpt4o_azure,
     df=bank_data,
     verbose=False,
     agent_type=AgentType.OPENAI_FUNCTIONS,