Thanh Vinh Vo commited on
Commit
4b82152
·
1 Parent(s): dc4500c
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -18,6 +18,7 @@ from smolagents import (
18
  ToolCollection,
19
  VisitWebpageTool,
20
  GoogleSearchTool,
 
21
  )
22
  import whisper
23
 
@@ -182,7 +183,7 @@ class BasicAgent:
182
  def __init__(self):
183
  print("BasicAgent initialized.")
184
  self.multimodal_agent = CodeAgent(
185
- tools=[VisitWebpageTool(), GoogleSearchTool("serper"), get_file, audio_to_text],
186
  model= OpenAIServerModel(model_id="gpt-4o"),
187
  additional_authorized_imports=[
188
  "requests",
@@ -210,7 +211,6 @@ class BasicAgent:
210
  - Writing code to solve problem.
211
  - Browse the web to find information.
212
  - Reason across audio, vision, and text, a.k.a multimodal agent. """,
213
- verbosity_level=0,
214
  max_steps=10,
215
  )
216
 
@@ -229,15 +229,11 @@ class BasicAgent:
229
  "pandas",
230
  "io",
231
  "PIL",
232
- "chess",
233
  "img2text",
234
- "chess.pgn",
235
  "PIL.Image",
236
  "bytes",
237
  "cv2",
238
  "numpy",
239
- "chess.engine",
240
- "whisper",
241
  "openpyxl"
242
  "json",
243
  ],
 
18
  ToolCollection,
19
  VisitWebpageTool,
20
  GoogleSearchTool,
21
+ WikipediaSearchTool,
22
  )
23
  import whisper
24
 
 
183
  def __init__(self):
184
  print("BasicAgent initialized.")
185
  self.multimodal_agent = CodeAgent(
186
+ tools=[VisitWebpageTool(), GoogleSearchTool("serper"), get_file, audio_to_text, WikipediaSearchTool()],
187
  model= OpenAIServerModel(model_id="gpt-4o"),
188
  additional_authorized_imports=[
189
  "requests",
 
211
  - Writing code to solve problem.
212
  - Browse the web to find information.
213
  - Reason across audio, vision, and text, a.k.a multimodal agent. """,
 
214
  max_steps=10,
215
  )
216
 
 
229
  "pandas",
230
  "io",
231
  "PIL",
 
232
  "img2text",
 
233
  "PIL.Image",
234
  "bytes",
235
  "cv2",
236
  "numpy",
 
 
237
  "openpyxl"
238
  "json",
239
  ],