Thanh Vinh Vo
commited on
Commit
·
4b82152
1
Parent(s):
dc4500c
update
Browse files
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 |
],
|