Update app.py

app.py

@@ -320,7 +320,18 @@ def main():
     model_options = ["gpt-4o", "gpt-4o-mini"]
     selected_model = st.selectbox("Choose a GPT model", model_options)
     
-    llm = ChatOpenAI(
+    # llm = ChatOpenAI(
+    #     model=selected_model,#"gpt-4o-mini",
+    #     temperature=0,
+    #     max_tokens=None,
+    #     timeout=None,
+    #     max_retries=2,
+    #     api_key=os.getenv("OPENAI_API_KEY"),  # if you prefer to pass api key in directly instaed of using env vars
+    #     # base_url="...",
+    #     # organization="...",
+    #     # other params...
+    # )
+    llm = OpenAI(
         model=selected_model,#"gpt-4o-mini",
         temperature=0,
         max_tokens=None,