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")
                              )
             
@@ -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,
@@ -378,8 +378,8 @@ class KeyValueExtractor:
 
     def gradio_interface(self):
 
-        with gr.Blocks(css="style.css",theme="freddyaboulton/test-blue") as app:
-              gr.HTML("""<div><img src="https://www.broadridge.com/_assets/images/logos/br-pimary-blue-logo.svg" alt="Broadridge" style="float:left;width:200px;height:50px;"></div>""")     
+        with gr.Blocks(css="style.css",theme='SherlockRamos/Feliz') as app:
+              gr.HTML("""<div><center><img src="https://www.broadridge.com/_assets/images/logos/br-pimary-blue-logo.svg" alt="Broadridge" style="float:left;width:200px;height:50px;"></center></div>""")     
 
               gr.HTML("""<center class="darkblue" text-align:center;padding:30px;'><center>
               <center><h1 class ="center" style="color:#fff"></h1></center>