yinde commited on
Commit
9674732
·
verified ·
1 Parent(s): 1c66c1a

Update app.py

Browse files

adding back the object detection tool in the correct way.

Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -119,15 +119,13 @@ custom_role_conversions=None,
119
  # Import tool from Hub
120
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
121
 
122
- #Initialise the simple object detection tool
123
- object_detection = simple_object_detection("00000001_011.jpg", 0.2)
124
 
125
  with open("prompts.yaml", 'r') as stream:
126
  prompt_templates = yaml.safe_load(stream)
127
 
128
  agent = CodeAgent(
129
  model=model,
130
- tools=[DuckDuckGoSearchTool(), final_answer], ## add your tools here (don't remove final answer)
131
  max_steps=6,
132
  verbosity_level=1,
133
  grammar=None,
 
119
  # Import tool from Hub
120
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
121
 
 
 
122
 
123
  with open("prompts.yaml", 'r') as stream:
124
  prompt_templates = yaml.safe_load(stream)
125
 
126
  agent = CodeAgent(
127
  model=model,
128
+ tools=[DuckDuckGoSearchTool(), simple_object_detection, final_answer], ## add your tools here (don't remove final answer)
129
  max_steps=6,
130
  verbosity_level=1,
131
  grammar=None,