Update app.py

app.py

@@ -53,17 +53,9 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
 with open("prompts.yaml", 'r') as stream:
     prompt_templates = yaml.safe_load(stream)
     
-agent = CodeAgent(
-    model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
-    max_steps=6,
-    verbosity_level=1,
-    grammar=None,
-    planning_interval=None,
-    name=None,
-    description=None,
-    prompt_templates=prompt_templates
-)
+from google.colab import userdata
+model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=userdata.get('GEMINI_API_KEY'))
+agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=model)
 
 
 GradioUI(agent).launch()