Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import os
 from tools.final_answer import FinalAnswerTool
 from tools.web_search import DuckDuckGoSearchTool
 from tools.visit_webpage import VisitWebpageTool
@@ -65,9 +66,10 @@ visit_webpage = VisitWebpageTool()
 
 
 model = LiteLLMModel(
-num_ctx=8192,
-model_id='ollama_chat/gemma3:4b',
-api_base='http://127.0.0.1:11434',
+    model_id="gemini/gemini-1.5-flash-latest",
+    api_key=os.getenv("GEMINI_KEY"),
+    max_tokens=4096,
+    temperature=0.6
 )
 
 # Import tool from Hub