Update app.py

app.py

@@ -45,15 +45,9 @@ class BasicAgent:
             describe_image,
             read_file,
             transcribe_audio
-        ]
-        self.llm = AzureChatOpenAI(
-            model_name="gpt-4o",
-            api_key=self.get_access_token(),
-            azure_endpoint="https://cog-sandbox-dev-eastus2-001.openai.azure.com/",
-            api_version="2024-08-01-preview"
-        )
-        self.llm_with_tools = self.llm.bind_tools(self.tools)
+        ]        
         self.google_llm = ChatGoogleGenerativeAI(model='gemini-2.0-flash-lite')
+        self.llm_with_tools = self.google_llm.bind_tools(self.tools)
         self.graph = self.create_graph()        
 
     def __call__(self, task_id: str, question: str, file_name: str = None, local_file_name: str = None) -> str: