Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -5,7 +5,7 @@ from smolagents import VisitWebpageTool, WebSearchTool, WikipediaSearchTool, Pyt
|
|
| 5 |
class GaiaAgent:
|
| 6 |
def __init__(self):
|
| 7 |
self.model = HfApiModel(
|
| 8 |
-
|
| 9 |
temperature = 0.5,
|
| 10 |
model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
|
| 11 |
custom_role_conversations=None
|
|
|
|
| 5 |
class GaiaAgent:
|
| 6 |
def __init__(self):
|
| 7 |
self.model = HfApiModel(
|
| 8 |
+
max_tokens = 2096,
|
| 9 |
temperature = 0.5,
|
| 10 |
model_id = 'https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud',
|
| 11 |
custom_role_conversations=None
|