batgre commited on
Commit
6cd131e
·
verified ·
1 Parent(s): 00dc8a5

Revert ZeroGPUCodeAgent to CodeAgent

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,7 +57,7 @@ image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_co
57
  with open("prompts.yaml", 'r') as stream:
58
  prompt_templates = yaml.safe_load(stream)
59
 
60
- agent = ZeroGPUCodeAgent(
61
  model=model,
62
  tools=[final_answer], ## add your tools here (don't remove final answer)
63
  max_steps=6,
 
57
  with open("prompts.yaml", 'r') as stream:
58
  prompt_templates = yaml.safe_load(stream)
59
 
60
+ agent = CodeAgent(
61
  model=model,
62
  tools=[final_answer], ## add your tools here (don't remove final answer)
63
  max_steps=6,