Update agent.py
Browse files
agent.py
CHANGED
|
@@ -52,8 +52,7 @@ def build_agent():
|
|
| 52 |
model_kwargs={
|
| 53 |
"temperature": 0.7,
|
| 54 |
"max_new_tokens": 100
|
| 55 |
-
}
|
| 56 |
-
task="question-answering"
|
| 57 |
)
|
| 58 |
|
| 59 |
tools = [use_wikipedia_tool, use_search_tool]
|
|
|
|
| 52 |
model_kwargs={
|
| 53 |
"temperature": 0.7,
|
| 54 |
"max_new_tokens": 100
|
| 55 |
+
}
|
|
|
|
| 56 |
)
|
| 57 |
|
| 58 |
tools = [use_wikipedia_tool, use_search_tool]
|