Thanh Vinh Vo commited on
Commit
8ea1c3b
·
1 Parent(s): a0d4043
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,6 +18,7 @@ from smolagents import (
18
  Tool,
19
  ToolCollection,
20
  VisitWebpageTool,
 
21
  )
22
  import whisper
23
 
@@ -202,7 +203,7 @@ class BasicAgent:
202
  )
203
 
204
  self.code_agent = CodeAgent(
205
- tools=[VisitWebpageTool(), GoogleSearchTool("serper"), get_file, audio_to_text, extract_table_from_html],
206
  model=InferenceClientModel(
207
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
208
  ),
 
18
  Tool,
19
  ToolCollection,
20
  VisitWebpageTool,
21
+ WikipediaSearchTool
22
  )
23
  import whisper
24
 
 
203
  )
204
 
205
  self.code_agent = CodeAgent(
206
+ tools=[VisitWebpageTool(), GoogleSearchTool("serper"), get_file, audio_to_text, extract_table_from_html, WikipediaSearchTool()],
207
  model=InferenceClientModel(
208
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
209
  ),