Update app.py
Browse files
app.py
CHANGED
@@ -26,7 +26,11 @@ class BasicAgent:
|
|
26 |
)
|
27 |
self.agent = CodeAgent(
|
28 |
tools=[DuckDuckGoSearchTool(), suggest_outfit, define_coordinates, get_weather],
|
29 |
-
model=model
|
|
|
|
|
|
|
|
|
30 |
)
|
31 |
|
32 |
def __call__(self, question: str) -> str:
|
|
|
26 |
)
|
27 |
self.agent = CodeAgent(
|
28 |
tools=[DuckDuckGoSearchTool(), suggest_outfit, define_coordinates, get_weather],
|
29 |
+
model=model,
|
30 |
+
pipeline_kwargs={
|
31 |
+
"max_new_tokens": 512, # allow longer answers
|
32 |
+
"truncate": True # automatically trim too long prompts
|
33 |
+
}
|
34 |
)
|
35 |
|
36 |
def __call__(self, question: str) -> str:
|