alamanna commited on
Commit
fe2e3fc
·
verified ·
1 Parent(s): 900fdfe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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: