Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
11 |
# --- Define Agent ---
|
12 |
class SmolAgentWrapper:
|
13 |
def __init__(self):
|
14 |
-
self.model = TransformersModel(model_id="
|
15 |
self.tools = [DuckDuckGoSearchTool()] # You can add more tools here
|
16 |
self.agent = CodeAgent(model=self.model, tools=self.tools)
|
17 |
|
|
|
11 |
# --- Define Agent ---
|
12 |
class SmolAgentWrapper:
|
13 |
def __init__(self):
|
14 |
+
self.model = TransformersModel(model_id="google/flan-t5-base", use_chat_template=False)
|
15 |
self.tools = [DuckDuckGoSearchTool()] # You can add more tools here
|
16 |
self.agent = CodeAgent(model=self.model, tools=self.tools)
|
17 |
|