Thanh Vinh Vo
commited on
Commit
·
59c594c
1
Parent(s):
763ca02
update
Browse files
app.py
CHANGED
|
@@ -23,8 +23,11 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
| 23 |
class BasicAgent:
|
| 24 |
def __init__(self):
|
| 25 |
print("BasicAgent initialized.")
|
|
|
|
|
|
|
|
|
|
| 26 |
self.agent = CodeAgent(
|
| 27 |
-
tools=[],
|
| 28 |
model=InferenceClientModel(),
|
| 29 |
additional_authorized_imports=["requests", "bs4"],
|
| 30 |
max_steps=10,
|
|
|
|
| 23 |
class BasicAgent:
|
| 24 |
def __init__(self):
|
| 25 |
print("BasicAgent initialized.")
|
| 26 |
+
html_parser_tool = load_tool(
|
| 27 |
+
"victor/dom-to-semantic-markdown", trust_remote_code=True
|
| 28 |
+
)
|
| 29 |
self.agent = CodeAgent(
|
| 30 |
+
tools=[html_parser_tool],
|
| 31 |
model=InferenceClientModel(),
|
| 32 |
additional_authorized_imports=["requests", "bs4"],
|
| 33 |
max_steps=10,
|