Update app.py

app.py

@@ -52,7 +52,7 @@ class KeyValueExtractor:
         """
         self.model = "facebook/bart-large-mnli"
         self.client = AzureOpenAI(api_key=os.getenv("AZURE_OPENAI_KEY"),  
-                            api_version="2023-07-01-preview",
+                            api_version="2024-02-01",
                             azure_endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
                              )
             
@@ -111,7 +111,7 @@ class KeyValueExtractor:
 
       # Load the summarization chain using the ChatOpenAI language model
       chain = load_summarize_chain(
-          llm = AzureChatOpenAI(azure_deployment = "GPT-3"),
+          llm = AzureChatOpenAI(azure_deployment = "GPT-4o"),
           chain_type="refine",
           question_prompt=prompt,
           refine_prompt=refine_prompt,
@@ -135,7 +135,7 @@ class KeyValueExtractor:
         
         # Call OpenAI GPT-3.5-turbo
         chat_completion = self.client.chat.completions.create(
-            model = "GPT-3",
+            model = "GPT-4o",
             messages = conversation,
             max_tokens=1000,
             temperature=0
@@ -173,7 +173,7 @@ class KeyValueExtractor:
             
             # Call OpenAI GPT-3.5-turbo
             chat_completion = self.client.chat.completions.create(
-                model = "GPT-3",
+                model = "GPT-4o",
                 messages = conversation,
                 max_tokens=1000,
                 temperature=0
@@ -339,7 +339,7 @@ class KeyValueExtractor:
 
         # Step 6: Load the summarization chain with Azure ChatGPT
         chain = load_summarize_chain(
-            llm=AzureChatOpenAI(azure_deployment="GPT-3"),
+            llm=AzureChatOpenAI(azure_deployment="GPT-4o"),
             chain_type="refine",
             question_prompt=prompt,
             refine_prompt=refine_prompt,