danielkorat commited on
Commit
6cab4b5
·
verified ·
1 Parent(s): b284fd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -1,5 +1,5 @@
1
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool
2
- from smolagents.default_tools import FinalAnswerTool
3
  from Gradio_UI import GradioUI
4
 
5
  import datetime
@@ -32,6 +32,7 @@ custom_role_conversions=None,
32
  final_answer = FinalAnswerTool()
33
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
34
  search_tool = DuckDuckGoSearchTool()
 
35
 
36
  with open("prompts.yaml", 'r') as stream:
37
  prompt_templates = yaml.safe_load(stream)
@@ -48,7 +49,8 @@ agent = CodeAgent(
48
  get_joke,
49
  get_random_fact,
50
  search_wikipedia,
51
- get_current_time_in_timezone
 
52
  ],
53
  max_steps=10,
54
  verbosity_level=4,
 
1
  from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool
2
+ from smolagents.default_tools import FinalAnswerTool, VisitWebpageTool
3
  from Gradio_UI import GradioUI
4
 
5
  import datetime
 
32
  final_answer = FinalAnswerTool()
33
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
34
  search_tool = DuckDuckGoSearchTool()
35
+ visit_webpage_tool = VisitWebpageTool()
36
 
37
  with open("prompts.yaml", 'r') as stream:
38
  prompt_templates = yaml.safe_load(stream)
 
49
  get_joke,
50
  get_random_fact,
51
  search_wikipedia,
52
+ get_current_time_in_timezone,
53
+ visit_webpage_tool
54
  ],
55
  max_steps=10,
56
  verbosity_level=4,