magdap116 commited on
Commit
655a207
·
verified ·
1 Parent(s): f3f3e4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from smolagents import DuckDuckGoSearchTool,HfApiModel,load_tool,PythonInterpret
7
  import hashlib
8
  import json
9
  from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
10
-
11
  import torch
12
  from tooling import ModelMathTool
13
 
@@ -105,6 +105,8 @@ class BasicAgent:
105
  verbosity_level=1,
106
  grammar=None,
107
  planning_interval=1,
 
 
108
 
109
  )
110
 
 
7
  import hashlib
8
  import json
9
  from transformers import AutoTokenizer, AutoModelForCausalLM , pipeline, TransformersEngine
10
+ import wikipedia
11
  import torch
12
  from tooling import ModelMathTool
13
 
 
105
  verbosity_level=1,
106
  grammar=None,
107
  planning_interval=1,
108
+ add_base_tools=True,
109
+ additional_authorized_imports=['requests','wikipedia']
110
 
111
  )
112