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]
|