Update app.py

app.py

@@ -48,7 +48,7 @@ class BasicAgent:
 
     def search_tool(self, prompt: str) -> str:
         # Initialize Gemini model
-        llm = ChatGoogleGenerativeAI(model="gemini-pro")  # or "gemini-1.5-flash"
+        llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash")  # or "gemini-pro"
         
         # Define Google Search tool (replace with actual search API)
         def google_search(query: str) -> str: